71dZddlmZddlmZddlmZddlmZddlmZGdde Z Gd d e Z Gd d e Z Gd de Zy)zClasses to define how concept args are added to argparse. A PresentationSpec is used to define how a concept spec is presented in an individual command, such as its help text. ResourcePresentationSpecs are used for resource specs. )absolute_import)division)unicode_literals)util) info_holderscPeZdZdZ ddZedZedZdZdZ y) PresentationSpeca5Class that defines how concept arguments are presented in a command. Attributes: name: str, the name of the main arg for the concept. Can be positional or flag style (UPPER_SNAKE_CASE or --lower-train-case). concept_spec: googlecloudsdk.calliope.concepts.ConceptSpec, The spec that specifies the concept. group_help: str, the help text for the entire arg group. prefixes: bool, whether to use prefixes before the attribute flags, such as `--myresource-project`. required: bool, whether the anchor argument should be required. If True, the command will fail at argparse time if the anchor argument isn't given. plural: bool, True if the resource will be parsed as a list, False otherwise. group: the parser or subparser for a Calliope command that the resource arguments should be added to. If not provided, will be added to the main parser. attribute_to_args_map: {str: str}, dict of attribute names to names of associated arguments. hidden: bool, True if the arguments should be hidden. Nc ||_||_||_||_||_||_||_|j||_| |_ y)aInitializes a ResourcePresentationSpec. Args: name: str, the name of the main arg for the concept. concept_spec: googlecloudsdk.calliope.concepts.ConceptSpec, The spec that specifies the concept. group_help: str, the help text for the entire arg group. prefixes: bool, whether to use prefixes before the attribute flags, such as `--myresource-project`. This will match the "name" (in flag format). required: bool, whether the anchor argument should be required. flag_name_overrides: {str: str}, dict of attribute names to the desired flag name. To remove a flag altogether, use '' as its rename value. plural: bool, True if the resource will be parsed as a list, False otherwise. group: the parser or subparser for a Calliope command that the resource arguments should be added to. If not provided, will be added to the main parser. hidden: bool, True if the arguments should be hidden. N) name _concept_spec group_helpprefixesrequiredpluralgroup_GetAttributeToArgsMap_attribute_to_args_maphidden) selfr concept_specr rrflag_name_overridesrrrs Blib/googlecloudsdk/command_lib/util/concepts/presentation_specs.py__init__zPresentationSpec.__init__5sS:DI%D DODMDMDKDJ"&"="=#DDKc|jS)zThe ConceptSpec associated with the PresentationSpec. Returns: (googlecloudsdk.calliope.concepts.ConceptSpec) the concept spec. )r rs rrzPresentationSpec.concept_spec]s   rc|jS)z\The map of attribute names to associated args. Returns: {str: str}, the map. )rrs rattribute_to_args_mapz&PresentationSpec.attribute_to_args_mapfs  & &&rct)aRGenerate a ConceptInfo object for the ConceptParser. Must be overridden in subclasses. Args: fallthroughs_map: {str: [googlecloudsdk.calliope.concepts.deps. _FallthroughBase]}, dict keyed by attribute name to lists of fallthroughs. Returns: info_holders.ConceptInfo, the ConceptInfo object. NotImplementedErrorrfallthroughs_maps r _GenerateInfozPresentationSpec._GenerateInfoos  rct)zGenerate a map of attributes to primary arg names. Must be overridden in subclasses. Args: flag_name_overrides: {str: str}, the dict of flags to overridden names. Returns: {str: str}, dict from attribute names to arg names. r )rrs rrz'PresentationSpec._GetAttributeToArgsMap~s  r)FFNFNF) __name__ __module__ __qualname____doc__rpropertyrrr$rrrr r sS4#'&P   ' '  rr c>eZdZdZdZdZe ddZdZdZ y) ResourcePresentationSpeczGClass that specifies how resource arguments are presented in a command.c @|sy|jD]}|jjD]}|j|k(s.t dj |dj |jjDcgc]}|jc}ycc}w)NzmAttempting to override the name for an attribute not present in the concept: [{}]. Available attributes: [{}], )keysr attributesr ValueErrorformatjoin)rrattribute_name attributes r_ValidateFlagNameOverridesz3ResourcePresentationSpec._ValidateFlagNameOverridess  -224((33) >>^ + 4 <>,HJK=LM M 5Js7BcP|j|i}t|jjD]p\}}|t |jjdz k(}|j |j |j ||j|}|sb|||j <r|S)N) is_anchor)r7 enumerater r1len GetFlagNamer r)rrrir6r:r s rrz/ResourcePresentationSpec._GetAttributeToArgsMaps##$78!$"4"4"?"?@ 9s4--889A==i    ..$))%8$--d 04inn- A ! rNcR|xsi}||vr|j|S|dk(ry|r|Stj}|rl|jtjr'||t tjddzz }||zS||j j dddzz }||zS)aGets the flag name for a given attribute name. Returns a flag name for an attribute, adding prefixes as necessary or using overrides if an override map is provided. Args: attribute_name: str, the name of the attribute to base the flag name on. presentation_name: str, the anchor argument name of the resource the attribute belongs to (e.g. '--foo'). flag_name_overrides: {str: str}, a dict of attribute names to exact string of the flag name to use for the attribute. None if no overrides. prefixes: bool, whether to use the resource name as a prefix for the flag. is_anchor: bool, True if this it he anchor flag, False otherwise. Returns: (str) the name of the flag. projectN-_)getrPREFIX startswithr<lowerreplace)r5presentation_namerrr:prefixs rr=z$ResourcePresentationSpec.GetFlagNames(.3,, $ $^ 44"   [[F  % %dkk 2#C $4$56<< N "" #))+33C=CC N ""rc tj|j|j|j|j ||j |j|j|j S)a,Gets the ResourceInfo object for the ConceptParser. Args: fallthroughs_map: {str: [googlecloudsdk.calliope.concepts.deps. _FallthroughBase]}, dict keyed by attribute name to lists of fallthroughs. Returns: info_holders.ResourceInfo, the ResourceInfo object. )rrrr) r ResourceInfor rr rrrrrr"s rr$z&ResourcePresentationSpec._GenerateInfosV  $ $    ""{{jj{{  rct|t|sy|j|jk(xr|j|jk(xr|j|jk(xr|j |j k(xrj|j |j k(xrO|j|jk(xr4|j|jk(xr|j|jk(SNF isinstancetyper rr rrrrrrothers r__eq__zResourcePresentationSpec.__eq__ eT$Z (  II # (   !3!3 3 ( OOu// / ( MMU^^ + (15 u||0K ( MMU^^ +  (15 ekk0I  ( KK5<< ' )rNFF) r&r'r(r)r7r staticmethodr=r$rTr+rrr-r-s5O M !IM,1 # #D,)rr-ceZdZdZy)InvalidPresentationSpecErrorz$Error for invalid presentation spec.N)r&r'r(r)r+rrrYrYs,rrYc8eZdZdZdZe ddZdZdZy)!MultitypeResourcePresentationSpecz&A resource-specific presentation spec.c `i}|jjDcgc]}|jj|r|!}}t|dkDrst j |j rTdjd|D}td|d|j dt j|j d|jjD]N}|g|k(}|j|j |j ||j|}|s@|||j <P|Scc}w) Nr9r/c34K|]}|jyw)N)r ).0as r zKMultitypeResourcePresentationSpec._GetAttributeToArgsMap..s<|!qvv|sz Multitype resource has anchors [z] and positional name [z]. Multitype resource can only be non-positional or have a single anchor. Update multitype collections or change the presentation name to .)rrr:) r r1 IsLeafAnchorr<r IsPositionalr r4rYFlagNameFormatr=r) rrrr_ leaf_anchors anchor_namesr6r:r s rrz8MultitypeResourcePresentationSpec._GetAttributeToArgsMaps2#11<<;Gets the MultitypeResourceInfo object for the ConceptParser. Args: fallthroughs_map: {str: [googlecloudsdk.calliope.concepts.deps. _FallthroughBase]}, dict keyed by attribute name to lists of fallthroughs. Returns: info_holders.MultitypeResourceInfo, the ResourceInfo object. )rrr) rMultitypeResourceInfor rr rrrrr"s rr$z/MultitypeResourcePresentationSpec._GenerateInfo/sO  - -    ""{{jj rct|t|sy|j|jk(xr|j|jk(xr|j|jk(xr|j |j k(xrj|j |j k(xrO|j|jk(xr4|j|jk(xr|j|jk(SrNrOrRs rrTz(MultitypeResourcePresentationSpec.__eq__DrUrrV) r&r'r(r)rrWr=r$rTr+rrr[r[s/.!.IM,111@*)rr[N)r) __future__rrr googlecloudsdk.calliope.conceptsr(googlecloudsdk.command_lib.util.conceptsrobjectr r- ExceptionrYr[r+rrrpsU''1Akvk\`)/`)F-9- X)(8X)r