?~dZddlmZddlmZddlmZddlZddlZGddeZGdd eZ Gd d eZ y) zBA module that provides parsing utilities for each command example.)absolute_import)division)unicode_literalsNcLeZdZdZdZdZd dZd dZdZdZ d Z d d Z d Z y)CommandStringParserasObject which builds and returns all metadata about string and flags. Attributes: command_node: calliope._CommandCommon, The command object that has access to argparse. command_parser: The argparse object for command_node. example_string: The most recently parsed example command string. example_metadata: The metadata gotten from example_string. cH||_|jj|_yN) command_node_parsercommand_parser)selfr s 5lib/googlecloudsdk/core/util/command_string_parser.py__init__zCommandStringParser.__init__%s$D++33Dc||_t|_|j|}|sy|jj |d}|j }|jD]w\}}t||d}t|tr|j|||7t|tjr|j|||e|j|||y|jS)aGets relevant metadata from an example command string. Args: example_command_string: The example command string to be parsed. Returns: An ExampleCommandMetadata object containing the relevant metadata or None if there was a parsing error. NT) raise_error)example_stringExampleCommandMetadataexample_metadataget_parse_argsr parse_argsGetSpecifiedArgsDictitemsgetattr isinstancelist parse_list collections OrderedDict parse_dict parse_others)r example_command_string parse_inputmetadatafilteredkeyvalue tmp_values rparsezCommandStringParser.parse)s1D24D%%&<=K  ""--kt-LH,,.Hnn& U(C.i It $  3. i!8!8 9  3. )S%0'   rNcjt|j}|d}|d}|j||} |jj t |d| } |jj d| } | | krt|j} t |d} |jj| | | } | t| zdz }|j| |dz}|rdj||n|}t|||| |}|jj||r|jdzSy)aGets metadata from an example command string for a simple dict-type arg. It updates the already existing ExampleCommandMetadata object of the example string with the metadata. Args: dict_arg: The dictionary-type argument to collect metadata about. key: The name of the argument's attribute in the example string's namespace. value: The actual input representing the flag/argument in the example string (e.g --zone, --shielded-secure-boot). prev_stop: Optional. The index in the example string the flag was last seen. count: Optional. The number of times the flag has been seen in the example string. Returns: The index in the example string where parsing stopped if the argument is specified multiple times. r-- {key}_{count}r&countN)rrget_start_searchrindexstrfindlenrfindformatArgumentMetadataradd_arg_metadata stop_index)r dict_argr&r' prev_stopr0 dict_list first_pair last_pair start_search start_index next_start last_valuelast_value_startr: arg_valuescope arg_metadatas rr zCommandStringParser.parse_dictJs<*X^^%&I1J" I(( :L%%++C 1 ,> MK$$))$  Mk&&++D+>j l "jB&6,,- x|$j,,22:|3=?#c*o5q8j%%k*Q,?i>Co$$E$:e&c9e[&02l ,,\: &&** rc |jj|}t|dtjrt|dk(r>t |dkDr/t |dj}t |dj}|dd}|dd}|j|} |jjt|| } |jjd| } | | krt |j} |jjt|| | t t|zdz } |j| | dz} t|| || | }|jj|yd}t!|D]}||}|j#|||||dz}!yd}t!|D]}||}|j%|||||dz}!y)aGets metadata from an example command string for nested list arguments. This is a helper function for parse_list(). It updates the already existing ExampleCommandMetadata object of the example string with the metadata. Args: list_arg: The list-type argument to collect metadata about. key: The name of the argument's attribute in the example string's namespace. value: The actual input representing the flag/argument in the example string (e.g --zone, --shielded-secure-boot). rr-r+r,N)rr0rrrr5rrr1r2r3r4r6r8rr9ranger r)r rJr&r' flag_count first_dict last_dictstartstopr@rArBr:rErGr<ivals rrIz%CommandStringParser.parse_nested_lists$$**51J(1+{667 qS]Q.(1+++-. "++-. 1 a }Q,,U3 ))//E LI ((--dK@  $3t7J7J3Kj))//D <0:<#d)n%'() '' JqLA 'Y[(24  ..|<  z"A #ooc3y!A#F)# iZ !qkOOCeY!D !rc t|ts|j|}|jj t ||}|dk(r|j ||}|jjt ||}t|||||tt |zdz }|jj|yy)aGets metadata from an example string for non list-type/dict-type args. It updates the already existing ExampleCommandMetadata object of the example string with the metadata. Args: other_arg: The non list-type and non dict-type argument to collect metadata about. key: The name of the argument's attribute in the example string's namespace. value: The actual input representing the flag/argument in the example string (e.g --zone, --shielded-secure-boot). r+r-N) rboolr1rr4r3get_enum_valuer2r8r5rr9)r other_argr&r'r@rArGs rr!z CommandStringParser.parse_otherss" i &**51l'',,S^\Jk  '' <@ ''--c)nlKk%c9c;&1CI4G&G!&KMl ,,\: 'rc|j|d}t|trP|j|j |j dd|j ddg}|D] }||vs|} t||vr|j }|jd}|dk7r|d|jdn|jd}t|dkDr"dj|d dj n|dj }|S) ajGets the input value of an enum argument in the example string. This is a helper function for parse_others(). Args: enum_value: The namespace value of the argument in question. prev_stop: The index in the example string where parsing stopped. Returns: The actual input in the example string representing the argument's value. N-_z --r+= r-) rrr3lowerupperreplacestripr4splitr5join)r enum_valuer<unparsed_stringpossible_versionsversionarg_endarg_lists rrVz"CommandStringParser.get_enum_values)))*5O*c"%++-z/?/?/A%--c37%--c379'' o %*' :o-'--/o$$U+g:AR-/(7+11#6&,,S17:(ma6GCHHXab\*002RL&&( rcX|r)|jj||t|z}n'|jj|t|z}|dk(rR|jjj dd}|jj|}|t|zdz}|S)aGets the position to begin searching for an argument's value. This is a helper function for all the parse functions aside from parse(). Args: namespace_name: The value of the argument in namespace's specified_args dictionary. ('INSTANCE_NAMES:1', '--zone', etc) prev_stop: Optional. For recurring flags, where the flag was last seen. Returns: The index in the example string to begin searching. r+rZrYr-)rr4r5r namer`)r namespace_namer<r@ command_namecommand_name_starts rr1z$CommandStringParser.get_start_search#s"))..~yI.)*l))..~>.)*l r&&++33C=l..33LA'#l*;;Q>l rc|jjjdd}|j|}|dk(ry|t |zdz}t j ||d}|S)zGets a list of arguments in a string. (Note: It assumes '$' is not included in the string) Args: string: The string in question. Returns: A list of the arguments from the string. rZrYr+Nr-)r rkr`r4r5shlexrb)r stringrmrncommand_name_stop args_lists rrz"CommandStringParser.get_parse_argsDsr$$))11#s;L\2R +S->>A F#4#567I r)rN)r) __name__ __module__ __qualname____doc__rr)r rrIr!rVr1rrrrrs;4!B,)\2+h3Ej;<"HBrrc4eZdZdZdZdZdZdZdZdZ y) rzEncapsulates metadata about entire example command string. Attributes: argument_metadatas: A list containing the metadata for each argument in an example command string. cg|_yr )_argument_metadatasr s rrzExampleCommandMetadata.__init__es !Drc:|jj|y)zAdds an argument's metadata to comprehensive metadata list. Args: arg_metadata: The argument metadata to be added. N)r{append)r rGs rr9z'ExampleCommandMetadata.add_arg_metadatahs  ##L1rc2t|jdS)z:Returns the metadata for an entire example command string.c|jSr r:xs rz?ExampleCommandMetadata.get_argument_metadatas..rs!,,rr&)sortedr{r|s rget_argument_metadatasz-ExampleCommandMetadata.get_argument_metadatasps $**0F GGrct|tryt|jd}t|jd}t |t |k7ryt t |jD]}||||k7syyy)Nc|jSr rrs rrz/ExampleCommandMetadata.__eq__..ws rrc|jSr rrs rrz/ExampleCommandMetadata.__eq__..xs1<rsB I&' @&@F -:V-:`%"v%"r