dZddlmZmZddlZddlmZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZedzZeZddeeeffdZdZdZdZy)zKLibrary for generating argument specifications for current implementations.)AnyDictN)cli_treemutexhiddennamenode_id argumentsrequiredgrouptypechoices positional)r?*z...global_-returnc|tj|d}|syi}t|j}|r ||t<|S)zGenerates the argument specifications for the calliope cli command node. Args: command_node: calliope command node cli object. Returns: The argument specifications for the command node. N)rCommand_AddArgsToGroup constraints ARGUMENTS) command_node argument_treeargss =lib/googlecloudsdk/command_lib/meta/generate_argument_spec.pyGenerateArgumentSpecificationsr (sD!!,5,  - 11 2$ #M) cg}|jD]}|jrtgi}|jr d|t<|j r d|t <|jr d|t<ttj|t<t||t<|ts|jt|i|j r|jt#||jt%||S)zAdd the given arguments to the given arguments group spec. Args: arguments: iterable: calliope objects representing the arguments group. Returns: The list of arguments added to the group spec. T)r is_groupris_mutexMUTEX is_requiredREQUIRED is_hiddenHIDDENstruuiduuid4NODE_IDrappendGROUP is_positional_GetPositionalSpec _GetFlagSpec)r args_group_specargchild_args_group_specs rrr;s/  c ||("o '+e$ *.h' (,f%'*4::<'8G$)8)=I& y )'<=>  /45\#./#!$ r!c|j}|jtrt}n|jtrt}nd}|t |d}|j t t}t|i}|j|t<|j|t<|jr d|t<|jrt!|j|t"<t%t'j(|t*<|S)zGet the flag spec for the given flag. Args: flag: The calliope object representing the flag. Returns: The flag spec for the given flag. NT)r startswith FLAG_PREFIXSHORT_FLAG_PREFIXlenreplace UNDERSCOREHYPHENNAMEr TYPEr&r' is_globalGLOBALrlistCHOICESr*r+r,r-)flag flag_name flag_prefix flag_specs rr2r2Zsii)+&K-.#KKK(*+) F3)Y)II)D/(()H ^^If \\dll+Ig4::<()G r!c|jjttj }t |t di}|jtv}|r ||t<ttj|t<|S)zGet the positional spec for the given positional. Args: positional: The calliope object representing the positional. Returns: The positional spec for the given positional. T)rr<r>r=upperr? POSITIONALnargsOPTIONAL_NARGSr'r*r+r,r-)rpositional_namepositional_specpositional_requireds rr1r1yslOO++FJ?EEG/?J=/#((> 3OH ./' r!)N)__doc__typingrrr+googlecloudsdk.callioperr%r)r?r-rr'r/r@rDrKrMrBr=r>r9r:r*r rr2r1r!rrUsR ,         %   qj c3h&>>r!