gdZddlmZddlmZddlmZddlmZddlmZddlm Z Gdd e jZ dd Z d Z d Zd ZdZy)z?Utilities for handling YAML schemas for gcloud update commands.)absolute_import)division)unicode_literals) arg_utils)yaml_arg_schema) exceptionsceZdZdZy)NoFieldsSpecifiedErrorz7Raises when no arguments specified for update commands.N)__name__ __module__ __qualname____doc__2lib/googlecloudsdk/command_lib/util/apis/update.pyr r s?rr c |js tdt}t|jj D]}|j t||||!djt|S)aGets the fieldMask that is required for update api calls. Args: args: The argparse parser. spec: The CommandData class. mask_path: string, the dotted path of mask in the api method is_dotted: Boolean, True if the dotted path of the name is returned. Returns: A String, represents a mask specifying which fields in the resource should be updated. Raises: NoFieldsSpecifiedError: this error would happen when no args are specified. z4Must specify at least one valid parameter to update.,) GetSpecifiedArgsr set_GetSpecParams argumentsparamsupdate_GetMaskFieldsjoinsorted)argsspec mask_path is_dotted field_setparams r GetMaskStringr#ss    > @@e)dnn334e ^E4IFG5 &# $$rct}|j|s|S|jD]#}t|||}|s|j |%|S)aGets the fieldMask based on the yaml arg and the arguments specified. Args: param: yaml_arg_schema.YAMLArgument, the yaml argument added to parser args: parser_extensions.Namespace, user specified arguments mask_path: str, path to where update mask applies is_dotted: bool, True if the dotted path of the name is returned Returns: Set of fields (str) to add to the update mask )rIsApiFieldSpecified api_fields_ExtractMaskFieldadd)r"rrr r! api_field mask_fields rrr:sTe)  " "4 ( ##i"9iCJmmJ$ rc#K|D]?}t|tjrt|jD]}|<|Ayw)zRecursively yields all the params in the spec. Args: params: List of Argument or ArgumentGroup objects. Yields: All the Argument objects in the command spec. N) isinstancer ArgumentGrouprr)rr"ps rrrQsAe%667eoo.!/ k sAAct|jd}|jd}|r.d|vr|jd}|d|}dj||dS||S)aExtracts the api field name which constructs the mask used for request. For most update requests, you have to specify which fields in the resource should be updated. This information is stored as updateMask or fieldMask. Because resource and mask are in the same path level in a request, this function uses the mask_path as the guideline to extract the fields need to be parsed in the mask. Args: mask_path: string, the dotted path of mask in an api method, e.g. updateMask or updateRequest.fieldMask. The mask and the resource would always be in the same level in a request. api_field: string, the api field name in the resource to be updated and it is specified in the YAML files, e.g. displayName or updateRequest.instance.displayName. is_dotted: Boolean, True if the dotted path of the name is returned. Returns: String, the field name of the resource to be updated.. .additionalPropertiesN)lensplitindexr)rr)r levelapi_field_listrepeated_indexs rr'r'bsn, iooc" #%??3'./%++,BCn%o~6n 88N56* ++ %  rc:d}|j}|D]}t||s|cS|jra|j}d}t||r tj||j }|D]"}t||sdj ||cSy)zGets the dotted path of mask in the api method. Args: method: APIMethod, The method specification. Returns: String or None. ) updateMask fieldMaskNz{}.{})GetRequestTypehasattr request_fieldrGetFieldFromMessagetypeformat)methodpossible_mask_fieldsmessagemaskr=request_messages rGetMaskFieldPathrFs5 #F ! ! # %'#dw k#  ((MOw &!55g6CEEIT% $ '~~mT22% rN)T)r __future__rrr$googlecloudsdk.command_lib.util.apisrrgooglecloudsdk.corerErrorr r#rrr'rFrrrrKsKF&':@*@Z--@%8."!Dr