'dZddlmZddlmZddlmZddlZddlZddlmZddlm Z ddl m Z  dd Z dd Z dd Zdd Zd ZdZ ddZ ddZGddeZGddeZdZdZ ddZdZdZy)z+Utilities for updating primitive dict args.)absolute_import)division)unicode_literalsN) arg_parsers)base)yamlc tjdj|dj||tjtj ||dj|S)N --update-{}{}={}key_type value_typez%List of key-value pairs to set as {}.metavaractiontypehelprArgumentformatr UpdateActionArgDict flag_name long_namer r key_metavar value_metavars 4lib/googlecloudsdk/command_lib/util/args/map_util.py MapUpdateFlagrsX 9%nn[-8  % %   Z H 2 9 9) D  cBt||||||j|SN)rr)r AddToParsergrouprrr rrrs rAddMapUpdateFlagr&-s/ !  Kr c tjdj||tjtj |dj|S)N --remove-{}) element_typezList of {} to be removed.r)rrrrrArgList)rrr rs r MapRemoveFlagr+@sI 9%  % %   H 5 & - -i 8  r c>t||||j|S)N)r)r+r#)r%rrr rs rAddMapRemoveFlagr-Js" H+ Kr cntjdj|ddj|S)N --clear-{} store_truezRemove all {}.)rr)rrr)rrs r MapClearFlagr1Ps2 )$   " "9 - r c8t||j|S)N)r1r#)r%rrs rAddMapClearFlagr3Xs i + 7 7 >>r c tjdj|dj||tjtj ||dj|S)N--set-{}r r zNList of key-value pairs to set as {0}. All existing {0} will be removed first.rrrs r MapSetFlagr6\sY  "nn[-8  % %   Z H y  r cBt||||||j|Sr")r6r#r$s r AddMapSetFlagr8ps/ !  Kr ceZdZdZddZdZy)ArgDictWithYAMLOrEnvz(Interpret a YAML or .env file as a dict.Nc ||_||_yzInitialize an ArgDictFile. Args: key_type: (str)->str, A function to apply to each of the dict keys. value_type: (str)->str, A function to apply to each of the dict values. Nr selfr rs r__init__zArgDictWithYAMLOrEnv.__init__DM DOr c<i}tjj|s$tjdj ||j dr file_pathmap_dict map_file_dictkeyvalues r__call__zArgDictWithYAMLOrEnv.__call__sH 77>>) $  ) ) % , ,Y 7 &!**yAm ++ = D DY O  nnY/m ^^M *++ E L L   $))+ U  O c"#  //%(% hsm, O O--.A.H.H.MN N O  --#**51   sD<"E-<.E*-.FNN__name__ __module__ __qualname____doc__r?rVr rr:r:s0!-r r:ceZdZdZddZdZy) ArgDictFilez Interpret a YAML file as a dict.Nc ||_||_yr<r r=s rr?zArgDictFile.__init__r@r ctj|}i}tj|s$tjdj ||j D]F\}}|jr |j|}|jr |j|}|||<H|S#t$r%tjdj |wxYw#t$r%tjdj |wxYw)NrCrDrE) rrMrNrrIrrOr rPr)r>rQrSrRrTrUs rrVzArgDictFile.__call__sNN9-MH >>- (  ) ) C J J   $))+ U  O c"#  //%(% hsm, O O--.A.H.H.MN N O  --#**51   s4B,C,.C.D rWrXr]r rr_r_s(!r r_c|jdj|dt||dj|yN --{}-file FILE_PATHr zvPath to a local YAML file with definitions for all {0}. All existing {0} will be removed before the new {0} are added.)rrr) add_argumentrr:r%rrr rs rAddMapSetFileFlagWithYAMLOrEnvrhsB# j I Gyr c|jdj|dt||dj|yrc)rfrr_rgs rAddMapSetFileFlagrjs@# Z @ Gyr c|s|}|j}|jdj|}t|||||t ||||t |||t |||||t|||||y)aRAdd flags for updating values of a map-of-atomic-values property. Args: parser: The argument parser flag_name: The name for the property to be used in flag names long_name: The name for the property to be used in help text key_type: A function to apply to map keys. value_type: A function to apply to map values. ztOnly --update-{0} and --remove-{0} can be used together. If both are specified, --remove-{0} will be applied first.)rr )r N)add_mutually_exclusive_groupadd_argument_grouprr&r-r3r8rj)parserrrr rr%update_remove_groups rAddUpdateMapFlagsrps I  - - /%00 ?y 1  & 9xP%I. Y H Y Hr c ^|j}|jdj||jdj||jdj||jdj||jdj|dS)zGet the flags for updating this map and return their values in a dict. Args: flag_name: The base name of the flags args: The argparse namespace Returns: A dict of the flag values r5r r/r(rd)set_flag_valueupdate_flag_valueclear_flag_valueremove_flag_valuefile_flag_value)GetSpecifiedArgsgetr)rargsspecified_argss rGetMapFlagsFromArgsr{(s((*.&**:+<+,>y,IJ  r c|riS|r|S|r|S|s|rK|xsi}|xsg}|jDcic] \}}||vs ||}}}|j|xsi|S|Scc}}w)aDetermine the new map property from an existing map and parsed arguments. Args: old_map: the existing map set_flag_value: The value from the --set-* flag update_flag_value: The value from the --update-* flag clear_flag_value: The value from the --clear-* flag remove_flag_value: The value from the --remove-* flag file_flag_value: The value from the --*-file flag Returns: A new map with the changes applied. )rOupdate) old_maprrrsrtrurvkvnew_maps r ApplyMapFlagsr<s, I  +mG)/R ' N11rs2&' /($ .& ? 4&:6:z$&$N  BF$N("r