o €Ïñã@sšdZddlmZddlmZddlmZddlmZddlmZddl m Z ddl Z Gd d „d e j ƒZ Gd d „d e ƒZ Gd d„de ƒZGdd„deƒZdS)z2Classes for runtime handling of concept arguments.é)Úabsolute_import)Údivision)Úunicode_literals)Ú parser_errors)Úutil)Ú exceptionsNc@seZdZdZdS)ÚErrorz%Base class for errors in this module.N)Ú__name__Ú __module__Ú __qualname__Ú__doc__©r r úF/tmp/google-cloud-sdk/lib/googlecloudsdk/calliope/concepts/handlers.pyrsrcó eZdZdZ‡fdd„Z‡ZS)Ú ParseErrorz#Raised if a concept fails to parse.cs d ||¡}tt|ƒ |¡dS)NzError parsing [{}]. {})ÚformatÚsuperrÚ__init__)ÚselfÚpresentation_nameÚmessageÚmsg©Ú __class__r rr"s zParseError.__init__©r r r r rÚ __classcell__r r rrrórcr)ÚRepeatedConceptNamezGRaised when adding a concept if one with the given name already exists.csd |¡}tt|ƒ |¡dS)NzRepeated concept name [{}].)rrrr)rÚ concept_namerrr rr*s zRepeatedConceptName.__init__rr r rrr'rrc@sBeZdZdZdd„Zdd„Zddd„Zd d „Zd d „Zd d„Z dS)ÚRuntimeHandlerzäA handler to hold information about all concept arguments in a command. The handler is assigned to 'CONCEPTS' in the argparse namespace and has an attribute to match the name of each concept argument in lower snake case. cCsd|_i|_g|_dS©N)Ú parsed_argsÚ_arg_name_lookupÚ _all_concepts©rr r rr6s zRuntimeHandler.__init__cCs|jS)z=Basically a lazy property to use during lazy concept parsing.)r!r$r r rÚ ParsedArgs<szRuntimeHandler.ParsedArgsTcsŠddlm‰G‡‡‡fdd„dtƒ}t|ˆƒrtˆƒ‚t|ˆ||j|jƒƒ|j  ˆ|ˆdœ¡t   |j ¡D] \}}||j t |¡<q6dS)aŒAdds a concept handler for a given concept. Args: name: str, the name to be used for the presentation spec. concept_info: ConceptInfo, the object that holds dependencies of the concept. required: bool, True if the concept must be parseable, False if not. Raises: RepeatedConceptName: If the given "name" has already been used with a concept. r)Úconceptscs(eZdZdZdd„Z‡‡‡fdd„ZdS)z,RuntimeHandler.AddConcept..LazyParsezBClass provided when accessing a concept to lazily parse from args.cSs||_||_dSr )ÚparseÚ arg_getter)rr'r(r r rrTs z5RuntimeHandler.AddConcept..LazyParse.__init__c sNz| | ¡¡WSˆjy&}zˆrtˆt |¡ƒ‚WYd}~dSd}~wwr )r'r(ÚInitializationErrorrÚsixÚ text_type)rÚe©r&ÚnameÚrequiredr rÚParseXs€ýz2RuntimeHandler.AddConcept..LazyParse.ParseN)r r r r rr0r r-r rÚ LazyParseQsr1)r.Ú concept_infor/N)Ú googlecloudsdk.calliope.conceptsr&ÚobjectÚhasattrrÚsetattrr0r%r#Úappendr*Ú iteritemsÚattribute_to_args_mapr"rÚNormalizeFormat)rr.r2r/r1Ú_Úarg_namer r-rÚ AddConcept@s  ýÿzRuntimeHandler.AddConceptcCs|j t |¡¡Sr )r"Úgetrr:)rr<r r rÚArgNameToConceptInfoksz#RuntimeHandler.ArgNameToConceptInfocCs|jD]}|d ¡qdS)Nr2)r#Ú ClearCache)rÚconcept_detailsr r rÚResetns ÿzRuntimeHandler.ResetcCs,zt||ƒWStyt d |¡¡‚w)a7Returns the value of the argument registered for dest. Based on argparse.Namespace.GetValue(). Args: dest: The dest of a registered argument. Raises: UnknownDestinationException: If no arg is registered for dest. Returns: The value of the argument registered for dest. z/No registered concept arg for destination [{}].)ÚgetattrÚAttributeErrorrÚUnknownDestinationExceptionr)rÚdestr r rÚGetValuers  ÿÿzRuntimeHandler.GetValueN)T) r r r r rr%r=r?rBrGr r r rr/s + r)r Ú __future__rrrÚgooglecloudsdk.callioperr3rÚgooglecloudsdk.corerr*rrrr4rr r r rÚs