dZddlmZddlZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lmZdd lmZ dd ZGd deZy)z,Provides a parser for --container arguments.) annotationsN)Sequence)Any)base)cli)parser_arguments) parser_errors)parser_extensionsflagsctjj||j|t|j|}|j |j_y)zAddContainerFlags updates parser to add --container arg parsing. Args: parser: The parser to patch. container_arg_group: Arguments that can be specified per-container. N)r ContainerFlag AddToParserContainerParserparserParseKnownArgsparse_known_args)rcontainer_arg_groupcontainer_parsers 6lib/googlecloudsdk/command_lib/run/container_parser.pyAddContainerFlagsrsM##F+!!&)$V]]4GH#3#B#B&-- cTeZdZdZdZdZ d dZd dZd dZd dZ ddZ y )rzIContainerParser adds custom container parsing behavior to ArgumentParser.z --containerz--presetcx|j|_|j|_|j|_||_y)zContainerParser constructor. Args: parser: The original command's parser. Used to parse non-container args. container_arg_group: Arguments to add to per-container parsers. N)r_parse_known_argsprog_prog_calliope_command_container_arg_group)selfrrs r__init__zContainerParser.__init__5s3$44DDJ#55D 3Drc|jg}g}|rd|j}t|tjr|j |j n|j|j|rdt|S)zE_GetContainerFlags returns the configured set of per-container flags.) rpop isinstance calliope_base ArgumentGroupextend argumentsappendname frozenset)r args flag_namesargs r_GetContainerFlagsz"ContainerParser._GetContainerFlagsEsg  % % &DJ  HHJc C44 5 CMM"#((#  Z  rctjd|j|j}t j |ddd}|j j|tjj||S)zD_NewContainerParser creates a new parser for parsing container args.F)add_helprcalliope_commandNT)r is_global cli_generatorallow_positional) r ArgumentParserrrrArgumentInterceptorrrrFLAG_INTERNAL_FLAG_FILE_LINE)r rais r_NewContainerParserz#ContainerParser._NewContainerParserQsr  - - ZZ//F  - -  B ))"-$$004 Mrc|jj|j}|r&tjddj |y)z_CheckForContainerFlags checks that no container flags were specified. Args: namespace: The namespace to check. zJWhen --container is specified {flags} must be specified after --container.z, r N)r/ intersectionGetSpecifiedArgNamesr ArgumentErrorjoin)r namespacecontainer_flagss r_CheckForContainerFlagsz'ContainerParser._CheckForContainerFlagsdsW --/<<&&(O  ' '  /* rcg}tjt}|}d}|t|kr||}|dz }||jk(r0|t|k\r|j |n|||}|dz }nt |tr9|j|jdzr||jddd}nTt |trq|j|jdzrS|jddd}|jdd} | tjvr|| }|j |nt |tr|j|jrl|t|kr^||}|jdd} | tjvr-|||}|j ||j |||dz }n<|dk(r&|j ||j||dn!|j ||t|kr|s|j||Si|_|j|j d <|j#D]I\} } t%j&} |j)j+| | } | |j| <K|j||\}} |j-||| fS) aHPerforms custom --container arg parsing. Groups arguments after each --container flag to be parsed into that container's namespace. For each container a new parser is used to parse that container's flags into fresh namespace and those namespaces are stored as a dict in namespace.containers. Remaining args are parsed by the orignal parser's parse_known_args method. Args: args: The arguments to parse. namespace: The namespace to store parsed args in. Returns: A tuple containing the updated namespace and a list of unknown args. r=)sepmaxsplit:z--N)r,r@ containers) collections defaultdictlistlen_CONTAINER_FLAG_NAMEr)r$str startswithsplit_PRESET_FLAG_NAMEr INGRESS_CONTAINER_PRESETSr'rrI_specified_argsitemsr Namespacer: parse_argsrB)r r,r@ remainingrIcurrentivalue preset_arg preset_namecontainer_namecontainer_argscontainer_namespace unknown_argss rrzContainerParser.ParseKnownArgsts(I((.JG A c$i-1ge1fa $++ + D >   5 !tAw'' q&! eS !e&6&6  # #c )'U[[S1[=a@A UC t55;<[[S1[5a8  &&s+A. %99 9{+' UC t556#d)m!W  &&s+A. %99 9tAw''   5 !   47 # Q D=ab" u[ c$i-^   # #i # HHI.2.G.GIl+*4*:*:*<&-779 446AA)<B.Ai>* +=#44 )5I|   + l ""rN)r parser_extensions.ArgumentParserrcalliope_base.ArgumentGroup)returnzfrozenset[str])rdrb)r@parser_extensions.Namespace)r,z Sequence[Any]r@rerdz1tuple[parser_extensions.Namespace, Sequence[Any]]) __name__ __module__ __qualname____doc__rNrRr!r/r:rBrrrrr/s\Q& 4.474 !& X# X#-X#9 X#rr)rz$parser_arguments.ArgumentInterceptorrrc)ri __future__rrJcollections.abcrtypingrgooglecloudsdk.callioperr%rrr r googlecloudsdk.command_lib.runr robjectrrjrrrqsN3"$9'4150 C 0 C4 C ]#f]#r