o  @sdZddlmZddlmZddlmZddlmZddlZddlmZddl Z ddl m Z dd l m Zdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZdZdZdZdZd\Z Z!d\Z"Z#Z$Z%d\Z&Z'Z(dZ)  dUddZ*Gddde+Z,Gddde,Z-Gdd d e.Z/d!d"Z0d#Z1e2d$Z3d%d&Z4e5e6e6e7e8ej9d'Z:d(d)Z;d*d+Zd0Z?Gd1d2d2e?ej@ZAe jBd3d4Gd5d6d6ZCGd7d8d8ej>d0ZDGd9d:d:ejEej>d0ZFGd;d<dZHGd?d@d@eFeDZIGdAdBdBeFeDZJGdCdDdDeFeDZKdEdFZLGdGdHdHejMZNdIdJZOGdKdLdLe?ejPZQGdMdNdNe?ejPZRe jBd3d4GdOdPdPZSGdQdRdRejEZTGdSdTdTe?ejUZVdS)Vz0Data objects to support the yaml command schema.)absolute_import) annotations)division)unicode_literalsN)Callable)Any)messages)actions) arg_parsers)arg_parsers_usage_text) arg_utils) labels_util) module_util) http_encoding__name____resource_id____relative_name____resource_type__keyvalue) arg_objectarg_dictarg_listarg_json)specparamsgroup file_typec Cs|r|}|p ||t<||t<||t<nt|i}||t<z |jdi|WStyM}z|j rGtd|j dd | |d}~ww)aTFormats a string with all the attributes of the given resource ref. Args: format_string: str, The format string. resource_ref: resources.Resource, The resource reference to extract attributes from. display_name: the display name for the resource. display_resource_type: Returns: str, The formatted string. zKKey [{}] does not exist. Must specify one of the following keys instead: {}rz, N) AsDictNameNAME_FORMAT_KEYRESOURCE_ID_FORMAT_KEY RelativeNameREL_NAME_FORMAT_KEYRESOURCE_TYPE_FORMAT_KEYformatKeyErrorargsjoinkeys) format_string resource_ref display_namedisplay_resource_typederrrrZ/tmp/google-cloud-sdk/lib/googlecloudsdk/command_lib/util/apis/yaml_command_schema_util.pyFormatResourceAttrStr2s(  r3c@eZdZdZdS)ErrorzBase class for module errors.Nr __module__ __qualname____doc__rrrr2r5Ur5c@r4)InvalidSchemaErrorz+Error for when a yaml command is malformed.Nr6rrrr2r;Zr:r;c@s:eZdZdZeddZeddZd ddZd d ZdS) HookaRepresents a Python code hook declared in the yaml spec. A code hook points to some python element with a module path, and attribute path like: package.module:class.attribute. If arguments are provided, first the function is called with the arguments and the return value of that is the hook that is used. For example: googlecloudsdk.calliope.arg_parsers:Duration:lower_bound=1s,upper_bound=1m cCs||}|r ||SdS)zGets the hook from the spec data. Args: data: The yaml spec key: The key to extract the hook path from. Returns: The Python element to call. N)getFromPath)clsdatarpathrrr2FromDataks  z Hook.FromDatacCs t|S)zGets the hook from the function path. Args: path: str, The module path to the hook function. Returns: The Python element to call. )ImportPythonHookGetHook)r?rArrr2r>{s z Hook.FromPathNcCs||_||_dSN) attributekwargs)selfrFrGrrr2__init__s z Hook.__init__cCs"|jdur|jdi|jS|jS)zaGets the Python element that corresponds to this hook. Returns: A Python element. Nr)rGrFrHrrr2rDs z Hook.GetHookrE) rr7r8r9 classmethodrBr>rIrDrrrr2r<_s   r<c Cs|d}t|dkrt|dkrtd|zt|dd|d}Wntjy<}ztd||d}~wwd}t|dkrri}|dd D]#}|sSqN|d }t|dkretd ||d||d<qNt||S) auImports the given python hook. Depending on what it is used for, a hook is a reference to a class, function, or attribute in Python code. Args: path: str, The path of the hook to import. It must be in the form of: package.module:attribute.attribute where the module path is separated from the class name and sub attributes by a ':'. Additionally, ":arg=value,..." can be appended to call the function with the given args and use the return value as the hook. Raises: InvalidSchemaError: If the given module or attribute cannot be loaded. Returns: Hook, the hook configuration. :zzInvalid Python hook: [{}]. Hooks must be in the format: package(.module)+:attribute(.attribute)*(:arg=value(,arg=value)*)?rz&Could not import Python hook: [{}]. {}N,=zKInvalid Python hook: [{}]. Args must be in the form arg=value,arg=value,...) splitlenr;r'r ImportModuleImportModuleErrorstripr<)rApartsattrerGarg arg_partsrrr2rCs8      rCstore_true_false)store store_trueappendcCsf|sdSt|tr|tkrtjS|tvr|St|S|d}|r+t j |fi|St dt|)aaParse the action out of the argument spec. Args: action: The argument action spec data. flag_name: str, The effective flag name. Raises: ValueError: If the spec is invalid. Returns: The action to use as argparse accepts it. It will either be a class that implements action, or it will be a str of a builtin argparse type. N deprecatedzUnknown value for action: ) isinstancestrTRUE_FALSE_ACTIONr StoreTrueFalseActionSTATIC_ACTIONSr<r>r=r DeprecationAction ValueError)action flag_name deprecationrrr2 ParseActions   rk)rbintlongfloatboolbytescCsf|tkr t|S|tkrtdS|tkrt|S|tkr$t |St |d}r.|St|S)zParses type from string. Args: arg_type: str, string representation of type data: dict, raw argument data Returns: The type to use as argparse accepts it. z,googlecloudsdk.calliope.arg_parsers:ArgList:N) ARG_OBJECT ArgObjectrBARG_LISTr<r>ARG_JSONArgJSON FILE_TYPE FileContents BUILTIN_TYPESr=)arg_typer@ builtin_typerrr2_ParseTypeFromStrs     r{cstfddtttfD}d}r|}n|rt}nd}|r5|tkr5d}td|d|d|s;|s;dSt|trIt |vrIt St|t rSt |Std t |) a Parse the action out of the argument spec. Args: data: dict, raw arugment data Raises: ValueError: If the spec is invalid. InvalidSchemaError: If spec and non arg_object type are provided. Returns: The type to use as argparse accepts it. c3s|]}|vVqdSrEr).0rr@rr2 szParseType..typeNarg_namezEOnly flags with type arg_object may contain a spec declaration. Flag z has type z1. Update the type or remove the spec declaration.zUnknown value for type: )anySPECPARAMSGROUPr=rqr;radictARG_DICTArgDictrBrbr{rg)r@ contains_specspecified_typeryrrr}r2 ParseType s,      rc@s$eZdZdZddZeddZdS)Choicez3Holds information about a single enum choice value.cCs`|d|_t|jtr|j|_d|vr|d|_nt|j|_|d|_|d|_ dS)N arg_value enum_value help_texthidden) rrarblowerrr ChoiceToEnumNamer=rr)rHr@rrr2rI4s     zChoice.__init__cCs|siSdd|DS)a-Converts a list of choices into a map for easy value lookup. Args: choices: [Choice], The choices. Returns: {arg_value: enum_value}, A mapping of user input to the value that should be used. All arg_values have already been converted to lowercase for comparison. cSi|]}|j|jqSr)rr)r|crrr2 Nz&Choice.ToChoiceMap..r)r?choicesrrr2 ToChoiceMap@s zChoice.ToChoiceMapN)rr7r8r9rIrKrrrrr2r1s  rc@eZdZdZejddZdS) TypeGeneratorz-Interface for generating a type from a field.cCdS)zGenerates a type from a field.NrrHfieldrrr2 GenerateTypeTzTypeGenerator.GenerateTypeN)rr7r8r9abcabstractmethodrrrrr2rQsr) metaclassc@$eZdZdZeddZddZdS)rwz0Holds information about a file content argument.cC~|S)z&Creates a FileContents from yaml data.rr?r@rrr2rB\szFileContents.FromDatacCsJ|r |jtjjkrtjddS|jtjjkrtjddStd|jd)NF)binaryTzConflicting types for field: zB. Only string and bytes fields are supported when using file_type.) variantapitools_messagesVariantSTRINGr rwBYTESr;namerrrr2rds   zFileContents.GenerateTypeNrr7r8r9rKrBrrrrr2rwYs   rwT)frozenc@s\eZdZUdZe  dddZded<ded <ded <d ed <d ed<d ed<ded<dS) _FieldSpeczCHolds information about a field and type that is generated from it.NFc Cs~|p|j}t|}||jkrtd|d|jt|tr%||} n|} ||||p-|| |j|dur9||dS|j|dS)aCreates a _FieldSpec from user input. If value is not provided in yaml schema by user, the value is defaulted to a value derived from the apitools field. Args: field: apitools field instance api_field: The name of the field under the repeated message that the value should be put. field_type: The type of the field if specified by user. arg_name: The name of the key in the dict. required: True if the key is required. hidden: True if the help text should be hidden. Returns: _FieldSpec instance Raises: ValueError: if the field contradicts the values provided by the user zExpected to receive field z but got N)r api_fieldr field_typerepeatedrequiredr) rr GetChildFieldNamergrarrrr) r?rrrrrr field_namechild_field_namegenerated_field_typerrr2 FromUserDatats.      z_FieldSpec.FromUserDatazapitools_messages.FieldrrbrrCallable[[str], Any] | Nonerrorrz bool | Noner)NNNNF)rr7r8r9rKr__annotations__rrrr2rps  / rc@r) EquitableTypeajWrapper that deteremines if two message fields of same type are equal. This is needed because the old message may contain ouptut only fields the user is not able to specify. For example, a message field may contain output only field `uid` that the user is not able to specify. Message(foo=bar) should still "match" existing Message(foo=bar, uid=baz). cCr)DChecks if new value matches existing value based on what user input.NrrHexisting_value new_valuerrr2MatchesrzEquitableType.MatchesN)rr7r8r9rrrrrrr2rsrcs6eZdZdZfddZddZejddZZ S)_FieldSpecTypeaWrapper that holds the arg type and information about the type. Interface allows users to parse string into arg_type and then parse value into correct apitools field. Attributes: field: apitools field instance api_field: str, name of the field where value should be mapped in message. arg_name: str, name of key in dict. repeated: bool, whether the field is repeated. required: bool, whether the field value is required. cs>tt|j|d|j|_|j|_|j|_|j|_|j|_dSN)ry)superrrIrrrrrrHry field_spec __class__rr2rIs  z_FieldSpecType.__init__cCs,|dur |jr g}n|}t||j|dS)zSets field in a message after value is parsed into correct type. Args: message_instance: apitools message instance we are parsing value into value: value we are parsing into apitools message N)rr SetFieldInMessager)rHmessage_instancer field_valuerrr2ParseIntoMessages z_FieldSpecType.ParseIntoMessagecCr)zBParses arg_value into apitools message using field specs provided.Nr)rHrrrr2__call__rz_FieldSpecType.__call__) rr7r8r9rIrrrr __classcell__rrrr2rs  rcs2eZdZdZd fdd ZddZddZZS) _FieldTypez~Type that converts string into apitools field instance. Attributes: choices: list[Choice], list of valid user inputs Nc  tt|jdi|||_dSNr)rrrIr)rHrrGrrr2rI z_FieldType.__init__cCs"||}tj|j||j|jdS)z*Converts string into apitools field value.)rr)ryr ConvertValuerrrrHrparsed_arg_valuerrr2rs  z_FieldType.__call__cCsx||krdS|dus|durdSt|tr|n|g}t|tr!|n|g}t|t|kr.dS|D] }||vr9dSq0dSrTNF)ralistrS)rHrr new_val_listexisting_val_listvalrrr2rsz_FieldType.MatchesrErr7r8r9rIrrrrrrr2rs rcCs(t||j}t||j}|||S)r)r GetFieldValueFromMessagerr)rrrexisting_field new_fieldrrr2_SubFieldMatches s rcs@eZdZdZfddZddZddZdd Zd d ZZ S) _MessageFieldTypezType that converts string input into apitools message. Attributes: field_specs: list[_FieldSpecType], list of message's fields c rr)rrrI field_specs)rHrrGrrr2rIrz_MessageFieldType.__init__cCs2|j}|jD]}||j}|||q|S)z)rz._MessageFieldType.__call__..)ryrarrrrrJr2r%s   z_MessageFieldType.__call__cs2|D]tfdd|jD}|rdSqdS)rc3s|] }t|VqdSrE)r)r|rnew_valrrr2r~0sz1_MessageFieldType._ContainsVal..TF)allr)rHr all_valuesmatchesrrr2 _ContainsVal-sz_MessageFieldType._ContainsValcCs|||krdS|dus|durdSt|tr|n|g}t|tr!|n|g}t|t|kr.dS|D] }|||s;dSq0dSr)rarrSr)rHrrrrrrrr2r7s" z_MessageFieldType.Matches) rr7r8r9rIrrrrrrrrr2rs  rc0eZdZdZfddZddZddZZS)_AdditionalPropsTypeaType converts string into list of apitools message instances for map field. Type function returns a list of apitools messages with key, value fields ie [Message(key=key1, value=value1), Message(key=key2, value=value2), etc]. The list of messages is how apitools specifies map fields. Attributes: key_spec: _FieldSpecType, specifes expected type of key field value_spec: _FieldSpecType, specifies expected type of value field c s&tt|jdi|||_||_dSr)rrrIkey_spec value_spec)rHrrrGrrr2rIXs z_AdditionalPropsType.__init__cCsX||}g}t|D]\}}|j}|j|||j||||q |SrE) rysorteditemsrrrrrr_)rHrrrkvrrrr2r]s   z_AdditionalPropsType.__call__cCs|||krdS|dus|durdSt|t|krdSdd|D}|D]}|j|vr-dS|j||j|js;dSq#dS)NTFcSrrr)r|rrrr2rrs z0_AdditionalPropsType.Matches..)rSrrrr)rHrr sub_field_maprrrr2rjs  z_AdditionalPropsType.Matchesrrrrr2rLs   rc@ eZdZdZddZddZdS) _MapFieldTypezCType converts string into apitools additional props field instance.cCs&||}|j}|j|||S)zBParses arg_value into additional props field of apitools messages.)ryrrr)rHradditional_props_fieldparent_messagerrr2rs  z_MapFieldType.__call__cCs.||krdS|dus|durdSt|||jSr)rryrrrr2rs z_MapFieldType.MatchesN)rr7r8r9rrrrrr2r}s rcCs$tj|j}|std|j|S)zReturns the input type for the apitools field. Args: field: apitools field instance Returns: Type function for apitools field input. Raises: InvalidSchemaError: if the field type is not listed in arg_utils.TYPES zUnknown type for field: )r TYPESr=rr;r)rryrrr2_GetFieldValueTypes rc@seZdZdZeddZeddZedddZedd d Z   dd d Z ddZ ddZ d ddZ d!ddZ d!ddZd ddZd"ddZdS)#rrz>A wrapper to bind an ArgObject argument to a message or field.cCs"|d}r|tkrt|SdS)!Creates ArgObject from yaml data.rN)r=rqr)r?r@ data_typerrr2_FieldTypeFromDataszArgObject._FieldTypeFromDatacCsHg}|D]}tj||p |d}|js|jr||jq||q|S)z,Creates ArgObject types from yaml spec data.) parent_field)rrrBrrextendr_)r? params_datarrr field_datarrrr2 _SpecFromDatas  zArgObject._SpecFromDataNcCs4|r|s|S|d}||r|t|dS|S)r.N) startswithrS)r?rrprefixrrr2_RelativeApiFields   zArgObject._RelativeApiFieldFc Cs|t}r |}n|}|d}|tp|t}dur'||||}nd}|d} |dsG|sG|dp;| } td| d| d||||| |d |d ||||d S) rrN json_namemutexrzapi_field is required for z: Add api_field to z to generate a valid ArgObject.rrrrrrrrdisable_key_description)r=rrrrr;rr) r?r@rrr group_datarrrrrrrr2rBs0   zArgObject.FromDatacCs.||_||_||_||_||_||_||_dSrEr)rHrrrrrrrrrr2rIs zArgObject.__init__cCs|jrtSdS)zReturns the correct argument action. Args: field: apitools field instance Returns: str, argument action string. r])rr FlattenActionrrrr2Actions zArgObject.Actioncs.t|jd}|jp g}tfdd|D|S)z4Returns first spec field that matches the api_field.)rc3s|] }|jkr|VqdSrErr|fr rr2r~sz2ArgObject._GetFieldTypeFromSpec..)rrrrnext)rHr default_typerrr r2_GetFieldTypeFromSpec s  zArgObject._GetFieldTypeFromSpeccCs&t||}||}|j||ddS)acRetrieves the the type of the field from messsage. Args: message: Apitools message class api_field: str, field path of message is_label_field: bool, whether field is part of labels map field Returns: _FieldSpecType, Type function that returns apitools message instance or list of instances from string value. F)is_label_fieldis_root)r GetFieldFromMessagerr)rHmessagerrrarg_objrrr2_GenerateSubFieldTypes zArgObject._GenerateSubFieldTypeTc Cs|j}rtd|d|jdz t|jjtj}Wntjy/tdj |jtjdw|j dk}t j ||j d}|j|jt|d}|j|jt|d}tj|||j|j||jd d } t| |||d } t| |d S) a6Returns function that parses apitools map fields from string. Map fields are proto fields with type `map<...>` that generate apitools message with an additionalProperties field Args: field_spec: _FieldSpec, information about the field is_root: whether the type function is for the root level of the message Returns: type function that takes string like 'foo=bar' or '{"foo": "bar"}' and creates an apitools message additionalProperties field Raises: InvalidSchemaError: if type is specified for a map field Field type z is not supported for map field (. Remove the type declaration from spec.zV{name} message does not contain field "{props}". Remove "{props}" from api field name.)rpropslabelsr)rT)key_type value_typerr root_levelrallow_key_onlyryrrrryr)rr;rr rrrADDITIONAL_PROPSUnknownFieldErrorr'rrrrrKEYVALUEr rrrrrrr) rHrrrrrprops_field_specrrradditional_prop_spec_typerrr2_GenerateMapType"s`    zArgObject._GenerateMapTypec sj}rtd|djdjdurddjD}nddhfd djjD}fd d|D}d d|D}tjd d |Dj |j j |j dd}t ||dS)aReturns function that parses apitools message fields from string. Args: field_spec: _FieldSpec, information about the field is_root: whether the _MessageFieldType is for the root level of the message Returns: _MessageFieldType that takes string like 'foo=bar' or '{"foo": "bar"}' and creates an apitools message like Message(foo=bar) or [Message(foo=bar)] Raises: InvalidSchemaError: if type is specified for a message field rz$ is not supported for message field rNcSsg|]}|jqSrr r rrr2rvsz2ArgObject._GenerateMessageType.. createTime updateTimecsg|] }|jvr|jqSr)rr )output_only_fieldsrr2rys  csg|] }jj|qSr)rrr)r|r)rrHrr2r}scSg|]}|jr|jqSrrrr rrr2rcSi|]}|j|qSrrr rrr2rz2ArgObject._GenerateMessageType..T)rr required_keysrrrrr ryrr)rr;rrrr all_fieldsr rrrrrrr)rHrrr field_namesrrrr)rr,rHr2_GenerateMessageType`s>       zArgObject._GenerateMessageTypec Cs|r|jtkrtj}tj}n|r|jtkrtj}tj}n |jp#t |j }d}t j ||j p-||j|jdt|t j |jd}t||ddS)aReturns _FieldType that parses apitools field from string. Args: field_spec: _FieldSpec, information about the field is_label_field: bool, whether or not the field is for a labels map field. If true, supplies default validation and help text. Returns: _FieldType that takes string like '1' or ['1'] and parses it into 1 or [1] depending on the apitools field type NF)rrrrrenable_file_uploadrryrr)rr%r KEY_FORMAT_VALIDATORKEY_FORMAT_HELPr&VALUE_FORMAT_VALIDATORVALUE_FORMAT_HELPrrrr rrrrrrarwrr)rHrrrdefault_help_textrrrr2_GenerateFieldTypes,   zArgObject._GenerateFieldTypecCsbtj||j|j|j|jd}t|}|tjj kr| ||S|tjj kr+| ||S| ||S)aGenerates a _FieldSpecType to parse the argument. Args: field: apitools field instance we are generating ArgObject for is_root: bool, whether this is the first level of the ArgObject we are generating for. is_label_field: bool, whether the field is for labels map field Returns: _FieldSpecType, Type function that returns apitools message instance or list of instances from string value. )rrrr)rrrrrrr GetFieldType FieldTypeMAPr)MESSAGEr6r>)rHrrrrfield_variationrrr2rs       zArgObject.GenerateTyperE)FN)NNNNNNF)F)T)TF)rr7r8r9rKrrrrBrIr rrr)r6r>rrrrr2rrs*    #     > 2%rrcCsFt||j}|jp t|}tj||j|j|jd}t ||| dS)zRetrieves the the type of the field from message. Args: message: Apitools message class user_field_spec: ArgDictFieldSpec, specifies the api field Returns: _FieldType, type function that returns apitools field class )rrrr8) r rrrrrrrrr ChoiceMap)ruser_field_specrryrrrr2_GetArgDictFieldTypes rFc@s4eZdZdZeddZddZddZdd Zd S) raA wrapper to bind an ArgDict argument to a message. The non-flat mode has one dict per message. When the field is repeated, you can repeat the message by repeating the flag. For example, given a message with fields foo and bar, it looks like: --arg foo=1,bar=2 --arg foo=3,bar=4 The Action method below is used later during argument generation to tell argparse to allow repeats of the dictionary and to append them. cCsr|d}|d}|dt}dd|dD}|dr3t|dkr'td t|||d |d d S||||S) NrrrcSsg|]}t|qSr)ArgDictFieldSpecrBr|r0rrr2rrz$ArgDict.FromData..rflattenrMz;Flattened ArgDicts must have exactly two items in the spec.rrOrrrr)r=rrSr;FlattenedArgDict)r?r@rrryfieldsrrr2rBs      zArgDict.FromDatacCs||_||_||_dSrE)rrrL)rHrrrLrrr2rIs zArgDict.__init__cCr)Nr_rrJrrr2r  szArgDict.Actioncs\tj|j|jd}fdd|jD}dd|D}tjdd|D|d}t|||dS) aGenerates an argparse type function to use to parse the argument. The return of the type function will be an instance of the given message with the fields filled in. Args: field: apitools field instance we are generating ArgObject for Raises: InvalidSchemaError: If a type for a field could not be determined. Returns: _MessageFieldType, The type function that parses the ArgDict and returns a message instance. rrcsg|]}tj|qSr)rFrr rrr2r r/z(ArgDict.GenerateType..cSr-rr.r rrr2r!r/cSr0rrr rrr2r$r1z(ArgDict.GenerateType..)rr2r3)rrrrrLr rr)rHrrrrrrrNr2rs  zArgDict.GenerateTypeN) rr7r8r9rKrBrIr rrrrr2rs  rc@r)rKaBA wrapper to bind an ArgDict argument to a message with a key/value pair. The flat mode has one dict corresponding to a repeated field. For example, given a message with fields key and value, it looks like: --arg a=b,c=d Which would generate 2 instances of the message: [{key=a, value=b}, {key=c, value=d}] cCs||_||_||_||_dSrErJ)rHrrrrrrr2rI8s zFlattenedArgDict.__init__cCsNtj||j|jd}t|j|j}t|j|j}tj ||d}t ||||dS)aGenerates an argparse type function to use to parse the argument. The return of the type function will be a list of instances of the given message with the fields filled in. Args: field: apitools field instance we are generating ArgObject for Raises: InvalidSchemaError: If a type for a field could not be determined. Returns: _AdditionalPropsType, The type function that parses the ArgDict and returns a list of message instances. rM)rrr!) rrrrrFrrrr rr)rHrrrrrrrr2r?s zFlattenedArgDict.GenerateTypeN)rr7r8r9rIrrrrr2rK,s rKc@sNeZdZUdZeddZded<ded<ded<d ed <d ed <d dZdS)rGaAttributes about the fields that make up an ArgDict spec. Attributes: api_field: The name of the field under the repeated message that the value should be put. arg_name: The name of the key in the dict. field_type: The argparse type of the value of this field. required: True if the key is required. choices: A static map of choice to value the user types. cCsH|d}|rdd|Dnd}||d|dt||dd|dS) NrcSsg|]}t|qSr)rrHrrr2rlr1z-ArgDictFieldSpec.FromData..rrrT)rrrrr)r=r)r?r@ data_choicesrrrr2rBis  zArgDictFieldSpec.FromDataz str | Nonerrrrrorzlist[Choice] | NonercCs t|jSrE)rrrrJrrr2rD{s zArgDictFieldSpec.ChoiceMapN)rr7r8r9rKrBrrDrrrr2rG\s    rGcr) _ArgJSONTypez>Parse json into apitools type but preserve arg_json help text.cs4tt|j|d|j|_|j|_t|j|_dSr)rrPrIrrr r?rrrrr2rIsz_ArgJSONType.__init__csTt|trjrfdd|DSjtjjkr"t|ts"t dt j j |S)Ncrr) _EncodeInput)r|rrJrr2rrz-_ArgJSONType._EncodeInput..z(Expecting map format i.e. {"foo": "bar"}) rarrrr r@JSONrr ArgumentTypeErrorEncodeToMessagerr)rHrrrJr2rQsz_ArgJSONType._EncodeInputc CsH||}z||WStjy#}z td|d|d}~ww)Nz Bad value [z]: )ryrQr rS)rHrrrYrrr2rs  z_ArgJSONType.__call__)rr7r8r9rIrQrrrrrr2rPs   rPc@r)ruz3A wrapper to bind an ArgJSON argument to a message.cCr)zCreates ArgJSON from yaml data.rrrrr2rBszArgJSON.FromDatacCsDt|}|tjjkr|tjjkrtdt|}tt |dS)NzbInvalid type: arg_json cannot be used for non-struct field types.Recommend changing to arg_object.r") r r?r@rR JSON_VALUEr;rrrPr ru)rHrrrrrr2rs    zArgJSON.GenerateTypeNrrrrr2rus   ru)NN)Wr9 __future__rrrrrcollections.abcr dataclassestypingrapitools.base.protorpcliterrgooglecloudsdk.callioper r r usage_text$googlecloudsdk.command_lib.util.apisr $googlecloudsdk.command_lib.util.argsr googlecloudsdk.corergooglecloudsdk.core.utilrr"r#r%r&r%r&rqrrsrtrrrrvr3 Exceptionr5r;objectr<rCrc frozensetrerkrbrlrnroEncoderxr{rrABCMetarFileTyperw dataclassrrDefaultArgTypeWrapperrrrrrrr ArgObjectTyperrrFRepeatedMessageBindableTyperrKrGrP ArgJSONTyperurrrr2s                 #7/! &  <)% :10C 0"