rxdZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l Z d Zd d Zd Zy )zCommon utility functions for ReplicaPool template processing. These utility functions enable easy replacement of parameters into ReplicaPool template files. )absolute_import)division)unicode_literals) arg_parsers) exceptions)yaml)filesNc|jdtjddd|jdtjddd y ) z)Adds --param and --param-from-file flags.z--param) min_lengthzA list of key=value parameters to substitute in the template before the template is submitted to the replica pool. This does not change the actual template file.z PARAM=VALUE)typehelpmetavarz--param-from-filezA list of files each containing a key=value parameter to substitute in the template before the template is submitted to the replica pool. This does not change the actual template file.zPARAM=FILE_PATHN) add_argumentrArgDict)parsers ;lib/googlecloudsdk/api_lib/compute/replica_template_util.pyAddTemplateParamArgsr!s`   ! , 3     ! ,  !c  |xsi}|xsi}t|}tj|D]:\}}||vrtjd|z t j |||<<tj|}t|trd|vrtjdt||\}}} |r0tjddj!t#|zt%|j'| z } | r0tjddj!t#| z|S#t j$r5}tjdj|||jd}~wwxYw) aParse and apply params into a template file. Args: template_file: The path to the file to open and parse. params: a dict of param-name -> param-value params_from_file: a dict of param-name -> param-file Returns: The parsed template dict Raises: yaml.Error: When the template file cannot be read or parsed. ArgumentError: If any params are not provided. ValidationError: if the YAML file is invalid. zDuplicate param key: z3Could not load param key "{0}" from file "{1}": {2}NtemplatezMInvalid template format. Root must be a mapping with single "template" valuezSSome parameters were present in the template but not provided on the command line: z, zWSome parameters were specified on the command line but not referenced in the template: )dictsix iteritemsrDuplicateErrorr ReadFileContentsError ArgumentErrorformatstrerrorr load_path isinstanceValidationErrorReplaceTemplateParamsjoinsortedsetkeys) template_fileparamsparams_from_file joined_paramskey file_pathermissing_params used_params unused_paramss r ParseTemplater34s  ? @@ // ;;+  $ $ ? F F9ajj * +++sD::F 0E==Fct|tjrj|jdrY|j drH|ddj }||vr||t t |gfS|t |gt fSt|trht }t }|jD]<}t|||\}}}|||<|j||j|>|||fSt|tret }t }g} |D]E} t| |\}}}| j||j||j|G| ||fS|t t fS)aApply the params provided into the template. Args: node: A node in the parsed template params: a dict of params of param-name -> param-value Returns: A tuple of (new_node, missing_params, used_params) where new_node is the node with all params replaced, missing_params is set of param references found in the node that were not provided and used_params were the params that we actually used. z{{z}}) r"r string_types startswithendswithstripr'rr(r$updatelistappend) noder*paramr0r1k new_subnode new_missingnew_usednew_nodesubnodes rr$r$gsoc&&' tt!41Rj e &u suc5'l33c5'lCE**dUN%K YY[-B q'6.*{Kd1gK("  .+ ..dUN%KH-B 6.*{Kook"K("  nk 22 su r)NN)__doc__ __future__rrrgooglecloudsdk.callioper"googlecloudsdk.command_lib.computergooglecloudsdk.corergooglecloudsdk.core.utilr rrr3r$rrrMs4 ''/9$* !&0f,r