K0dZddlmZddlmZddlmZddlmZddlmZddlmZddl m Z dd l Z Gd d e Z d Zy ) zConceptParsers manage the adding of all concept arguments to argparse parser. The ConceptParser is created with a list of all resources needed for the command, and they should be added all at once during calliope's Args method. )absolute_import)division)unicode_literals)deps)handlers)util)presentation_specsNcreZdZdZddZe ddZdZdZdZ dZ e d Z d Z d Zd Zd Zy) ConceptParserz4Class that handles adding concept specs to argparse.Nci|_g|_|D]}|j||j|xsi|_y)aUInitializes a concept holder. Args: specs: [presentation_specs.PresentationSpec], a list of the specs for concepts to be added to the parser. command_level_fallthroughs: {str: str}, a map of attributes to argument fallthroughs for those attributes. The format of the key should be FOO.a (the resource presentation name is "FOO" and the attribute name is "a"). The format of the value should either be "BAR.b" (where the argument depended upon is the main argument generated for attribute "b" of the resource presentation spec that is named "BAR"), or "--baz", where "--baz" is a non-resource argument that is added separately to the parser. Raises: ValueError: if two presentation specs have the same name or two specs contain positional arguments. N)_specs _all_args_AddSpec!_ValidateAndFormatFallthroughsMap_command_level_fallthroughs)selfspecscommand_level_fallthroughsspecs ?lib/googlecloudsdk/command_lib/util/concepts/concept_parsers.py__init__zConceptParser.__init__$sC&DKDN mmD'+'M'M"(b(*D$c tj|||||xsi||| | } i} t| || tj| xsiD]&\} }dj | j | }|| |<(|| g| S)axConstructs a ConceptParser for a single resource argument. Automatically sets prefixes to False. Args: name: str, the name of the main arg for the resource. resource_spec: googlecloudsdk.calliope.concepts.ResourceSpec, The spec that specifies the resource. group_help: str, the help text for the entire arg group. required: bool, whether the main argument should be required for the command. hidden: bool, whether or not the resource is hidden. 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. prefixes: bool, True if flag names will be prefixed with the resource name, False otherwise. Should be False for all typical use cases. 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. command_level_fallthroughs: a map of attribute names to lists of command- specific fallthroughs. These will be prioritized over the default fallthroughs for the attribute. Returns: (googlecloudsdk.calliope.concepts.concept_parsers.ConceptParser) The fully initialized ConceptParser. )requiredflag_name_overridespluralprefixesgrouphidden{}.{})r ResourcePresentationSpecUpdateFallthroughsMapsix iteritemsformatname)clsr& resource_spec group_helprrrrrrrpresentation_specfallthroughs_mapattribute_name fallthroughskeys r ForResourcezConceptParser.ForResource>sD+CC /52 *D2LM(+ "(b)*$ NN,11> Bc*s)* !"$4 55rc\tj|tj|k(ryy)a(Checks if two argument names match in the namespace. RESOURCE_ARG and --resource-arg will match with each other, as well as exact matches. Args: name: the first argument name. other_name: the second argument name. Returns: (bool) True if the names match. TF)rNormalizeFormat)rr& other_names r_ArgNameMatcheszConceptParser._ArgNameMatchesrs( D!T%9%9*%EE  rc|jD]}|j||jr%tdj ||jt j |sZt j |jsztdj ||jtj|jD]^\}}~t j|}||jvrtdj ||jj|`||j|j<y)aGAdds a given presentation spec to the concept holder's spec registry. Args: presentation_spec: PresentationSpec, the spec to be added. Raises: ValueError: if two presentation specs have the same name, if two presentation specs are both positional, or if two args are going to overlap. z:Attempted to add two concepts with the same name: [{}, {}]zFAttempted to add multiple concepts with positional arguments: [{}, {}]z0Attempted to add a duplicate argument name: [{}]N) r r3r& ValueErrorr%r IsPositionalr#r$attribute_to_args_mapr1rappend)rr* spec_nameaarg_namer&s rrzConceptParser._AddSpecs [[  i):)?)? @$$*F96G6L6L$MO O   I &   -22 3//5vi6G6L6L0NO O !}}%6%L%LM 8  ! !( +d  K &*, , nnD! N+zCinvalid fallthrough value: [{}]. Must be in the form BAR.b or --bazvalue)r#r$splitlenr5r%_ValidateSpecAndAttributeExist setdefault) rrspec_mapr.fallthroughs_listkeysr9r,fallthrough_stringvaluesvalue_spec_namevalue_attribute_names rrz/ConceptParser._ValidateAndFormatFallthroughsMaps H"%--0J"K  YYs^d Ta44:F3KA A#'i ))%NK 1 #))#. v;f $1178J1KM M v;! 28 ///  - -g.=.B D!2>)^D99VD1B,C%DEG HH ::>>) $D(,(9(9(D(DF(D9 (DFF ..4f#">>)^D$) /+ ,,FFs(C-c|jSN)r )rs rrzConceptParser.specss ;;rcr|jj}|s%tj}|j |t j |jD]X\}}|j|}|j||jtj|||jZy)z|Adds attribute args for all presentation specs to argparse. Args: parser: the parser for a Calliope command. )rN)dataconcept_handlerrRuntimeHandler add_conceptsr#r$r GetInfo AddToParser AddConceptrr1r)rparserruntime_handlerr9r concept_infos rr]zConceptParser.AddToParsers kk11O  //1o /*==5 4\\),lv&    y ) ==!"6rcg}|jD]7}|j|}|j}|s'|j|9d}dj t ||S)z:Returns a command line example arg string for the concept.c6|jdrdnd}||zS)Nz--ZA) startswith)argprefixs r_PositionalsFirstz._PositionalsFirstsnnT*sf c\r )r.)r r\GetExampleArgListextendrMrN)rexamplesr9infoargsris rGetExampleArgStringz!ConceptParser.GetExampleArgStrings^H[[ \\) $d  # # %d  !  88F8):; <NziInvalid fallthrough value [{}]: No argument associated with attribute [{}] in concept argument named [{}]zbad fallthrough string [{}]) rArBrArgFallthroughrrPr7r5r%)rrHrIr;r9r,rs r_MakeFallthroughzConceptParser._MakeFallthroughs  % %c *F 6{ah   ++ V "(i ZZ^^I &d++//Eh  <>  <<@@  "&&)c).."&=1DE0C,++,>?0CE)5 8N8N)Ny~~& ?  * *+; << EsC rV)FFNFFNN)__name__ __module__ __qualname____doc__r classmethodr/r3rrrCpropertyrr]rprsr\rrr r !si<*4AF48).9=1616f"rs8 ''151G x=Fx=v)r