ipSrSSKJr SSKJr SSKJr SSKrSSKJr SSKJ r SSK J r SS K Jr SS K Jr SS K Jr SS K Jr SS K Jr SSKJr SSKJr SSKJr "SS\5r"SS\5rSrSrSrSrSr Sr!Sr"Sr#Sr$Sr%Sr&S'S jr'"S!S"5r("S#S$\)5r*"S%S&\)5r+g)(z@Classes that generate and parse arguments for apitools messages.)absolute_import)division)unicode_literalsN)messages)base)util) arg_utils)update)yaml_arg_schema)yaml_command_schema)yaml_command_schema_util) labels_util) resources)resource_propertyc\rSrSrSrSrg)Error%z(Base class for this module's exceptions.N__name__ __module__ __qualname____firstlineno____doc____static_attributes__r;lib/googlecloudsdk/command_lib/util/apis/arg_marshalling.pyrr%s0rrc\rSrSrSrSrg)ConflictingResourcesError)zFError for whenever api method / primary resource cannot be determined.rNrrrrrr)sNrrcB[R"X5R$N)r GetFieldFromMessagetype)message api_fields r_GetLabelsClassr'-s  & &w : ? ??rct[X5n[R"X5n[R"XU5 gr")r'rParseCreateArgsr SetFieldInMessage)r%argsr& labels_cls labels_fields r_ParseLabelsIntoCreateMessager.1s,w2*,,T>, g,?rctX;d X(dSX'gSXRS5;agXS-S-UU'g)Nlabels,)split) static_fieldupdate_mask_paths r_AddLabelsToUpdateMaskr57sQ*  (%-L"  /55c:: '83>IrcURS5nUHn[X5nM U$![a [SRU55ef=f)N.z0The message does not have field specified in {}.)r2getattrAttributeErrorformat)r%r&path field_names r_RetrieveFieldValueFromMessager=Dsa  $jP,g . P  < C CI N PPPs )%Ac [RRU5nUR5(dg[ X5n[ X5nUR XT5nUR(a![R"XUR5 g)zHFind diff between existing labels and args, set labels into the message.FT) rDiffFromUpdateArgsMayHaveUpdatesr=r'Apply needs_updater r*r0)r%r+r&diffexisting_labels label_cls update_results r_ParseLabelsIntoUpdateMessagerHOsn    ( ( .$     27F/g1)**Y8- M4H4HI rc,[U[R5(aU/$[U[R5(a/$[U[R5(a UR n/nUHnUR [U55 M U$)zRetrieves all resource args from the arg_info tree. Args: params: an ArgGroup or list of args to parse through. Returns: YAMLConceptArgument (resource arg) list. ) isinstancer YAMLConceptArgumentArgument ArgumentGroup argumentsextend _GetResources)paramsresultparams rrPrP]s{;;<< 8O0011 I5566   F &e MM-&' -rcU(dgUVs/sHnURU5(dMUPM nnU(dGU(aURnURnOSnSn[R"SR XES95e[ U5S:aOUVs/sHo"RPM nn[R"SR SRU555eUS$s snfs snf)zRetrieves the primary resource arg. Args: resource_params: list of YAMLConceptParser resource_collection: registry.APICollection, resource collection associated with method Returns: YAMLConceptArgument (resource arg) or None. NaNo resource args were found that correspond with [{name} {version}]. Add resource arguments that corresponds with request.method collection [{name} {version}]. HINT: Can set resource arg is_primary_resource to True in yaml schema to receive more assistance with validation.)nameversionzOnly one resource arg can be listed as primary. Remove one of the primary resource args [{}] or set is_primary_resource to False in yaml schema., r) IsPrimaryResource full_name api_versionrInvalidSchemaErrorr:lenrUjoin)resource_paramsresource_collectionargprimary_resourcesrZr[primary_resource_namess r_GetPrimaryResourcerdts  %5$c  2 3 _5 %//i'33kik  ! ! "6"1  22 a2CD2C3hh2CD  ! ! vdii(>?@ BB 1 75*EsC$C$C)cUnU(d[US5=n(a [USS9/$/nUH2n[X%R5nUR[Xe55 M4 U$)aGets list of primary resource args and methods associated with them. Args: resource_args: list[YAMLConceptArg], list of potential primary resource args methods: list[registry.APIMethod], The method to generate arguments for. Returns: list[YAMLMethod] (resource arg) or None. Nprimary_resourcemethod)rdMethodResourceArgresource_argument_collectionappend) resource_argsmethodsr+primary yaml_methodsrh resource_args r_GetMethodResourceArgsrqsl $ !4T4!@@g@ wt D EE,f& 113L),?@ rcZUVs/sHn[R"U5PM sn$s snfr") resource_utilNormalizeFormat) attributesattrs r_NormalizeNamesrws%:D E*$- ' ' -* EE Es (cUHQnX :wdM URUR:XdM&[UR5[UR5:Hs $ g)a2Verify if there is a duplicated resource argument with the same attributes. Args: resource: yaml_arg_schema.Argument, resource to be verified. resource_params: [yaml_arg_schema.Argument], list to check duplicate. Returns: True if there is a duplicate resource arg in the list with same attributes. T)rUrwattribute_names)resourcer_res_args r%_DoesDupResourceArgHaveSameAttributesr|sP!gw||x}}<W44 5 X55 678 ! rc[5n[R"[5nUHnURnXA;a;XA;a5[ X05(d%[ R"SRU55eOURU5 URR5H'nXSR;dMX%RU5 M) M UR5VVs0sHupV[U5S:dMXV_M snn$s snnf)zRetrieves shared attributes between resource args. Args: resource_params: [yaml_arg_schema.Argument], yaml argument tree Returns: Map of attribute names to list of resources that contain that attribute. zxMore than one resource argument has the name [{}] with different attributes. Remove the duplicate resource declarations.rW)set collections defaultdictlistpresentation_namer|rr\r:addattribute_to_flag_mapvalues ignored_flagsrkitemsr])r_resource_namesflagsraarg_name flag_namerls r_GetSharedFlagsrs5.  ! !$ '% c$$H!  $7MM%% FFLfG  "..557 ++ + )8-:',kkm &3 ") ] a i&3  s #D<DcrURRnU(aURS5un nU$UnU$)Nr7)rjrZ rpartition)rh is_parentcollection_namer`_s r_GetCollectionNamersD77AA/ / : :3 ?A * rc$\rSrSrSrSrSrSrg)riiz4Method and the resource argument associated with it.cXlX lgr"rf)selfrgrhs r__init__MethodResourceArg.__init__ s ,Krc\UR(aURRU5$gr")rgParseResourceArgr namespaces rParseMethodResourceArg.Parses%   " " 3 3I >> r)rhrgN)rrrrrrrrrrrriris<rricN\rSrSrSrSrSrSrS SjrSr S r S r S r g) DeclarativeArgumentGeneratoria)An argument generator that operates off a declarative configuration. When using this generator, you must provide attributes for the arguments that should be generated. All resource arguments must be provided and arguments will only be generated for API fields for which attributes were provided. cDXl[UR5Ulg)zCreates a new Argument Generator. Args: arg_info: [yaml_arg_schema.Argument], Information about request fields and how to map them into arguments. N)arg_inforPrl)rrs rr%DeclarativeArgumentGenerator.__init__sM&t}}5Drc "^ [UR5n[U5nURVs/sHnUR X5PM nn[ URU5n[ SU55m UR5Hupx[[ U55n U RU 4SjS9 UR[R"USRSRU 5[R"U5S9S95 M U$s snf)zGenerates all the CLI arguments required to call this method. Args: methods: list[APIMethod], list of methods to generate arguments for. Returns: {str, calliope.base.Action}, A map of field name to the argument. c3t# UH.nUR=(a URRv M0 g7fr")rgrU).0ras r .7s1*(C :!5!5!:!::(s68c>UT;aS$U$)Nr)rU primary_namess r;DeclarativeArgumentGenerator.GenerateArgs..>s!62@D@r)keyzFor resources [{}], provides fallback value for resource {attr} attribute. When the resource's full URI path is not provided, {attr} will fallback to this flag value.rX)rvhelp)rrlrrGeneraterqr~rsortrkrrLr:r^rs StripPrefix) rrmshared_flag_resource_dictshared_resource_flag_listrar+primary_resource_argsrrlrrs @r GenerateArgs)DeclarativeArgumentGenerator.GenerateArgs's!00B0B C $%> ?}} &$ LL <$  &343E3EwO*(**M%>$C$C$E C ./n@B kk$-- DDJF99^,%11)<EKE> ?@ %F K+ &sD c/n[URU5nU(d [SSS9$[U5S:XaUR 5$UHnUR nUR nU(aU(d[R"S5e[XgRS9nURU5n U =(a U R5Rn X:XdMURU5 M [U5S:ad/n UH<nURU5=n (dMU RU R55 M> SR!U 5n [#SU S35e[U5S:XaUR 5$[SSS9$) aGets primary resource based on user input and returns single method. This determines which api method to use to make api request. If there is only one potential request method, return the one request method. Args: methods: list[APIMethod], The method to generate arguments for. namespace: The argparse namespace. Returns: MethodResourceArg, gets the primary resource arg and method the user specified in the namespace. Raises: ConflictingResourcesError: occurs when user specifies too many primary resources. NrfrWzIf more than one request collection is specified, a resource argument that corresponds with the collection, must be specified in YAML command.)rrXz5User specified multiple primary resource arguments: [z*]. Unable to determine api request method.)rqrlrir]poprhrgrr\ris_parent_resourcerGetCollectionInforZrk RelativeNamer^r)rrmrspecified_methodsrb method_inforhrgmethod_collectionspecified_resourceprimary_collectionurisparsedr+s rGetPrimaryResource/DeclarativeArgumentGenerator.GetPrimaryResourceJs&.t/A/A7K  T BB  1 $  " " $$( !!f$55 +%% )  - ??A',,Y7  ;  . . 0 : :  0  -/)2 ! d*+ &&y1 16 1 ++f))+ ,+YYt_d % A$H4 4 55 "  " " $$ T BBrNcUR5"5nU(a[R"XvU5nOUnU(aU[RR :Xa[ XUR5 O\U[RR:Xa>[XUR5n U (a![R"U5n [X:5 [R"XS9 URHn U RX(U5 M U$)aGenerates the request object for the method call from the parsed args. Args: namespace: The argparse namespace. method: APIMethod, api method used to make request message. static_fields: {str, value}, A mapping of API field name to value to insert into the message. This is a convenient way to insert extra data while the request is being constructed for fields that don't have corresponding arguments. labels: The labels section of the command spec. command_type: Type of the command, i.e. CREATE, UPDATE. existing_message: the apitools message returned from server, which is used to construct the to-be-modified message when the command follows get-modify-update pattern. Returns: The apitools message to be send to the method. ) static_fields)GetRequestTyper ParseExistingMessageIntoMessager CommandTypeCREATEr.r&UPDATErHr GetMaskFieldPathr5ParseStaticFieldsIntoMessagerr) rrrhrr0 command_typeexisting_message new_messager% need_updater4ras r CreateRequest*DeclarativeArgumentGenerator.CreateRequests2'')+K99 1gg ,88?? ?%g&:J:JK .::AA A3G4:4D4DF #44V<   A**7P}} ii+ NrcU(aU/O/nURXB5RU5n[RRUURR URR UR5S9$)aGets a resource reference for a resource returned by a list call. It parses the namespace to find a reference to the parent collection and then creates a reference to the child resource with the given id_value. Args: id_value: str, The id of the child resource that was returned. namespace: The argparse namespace. method: APIMethod, method used to make the api request Returns: resources.Resource, The parsed resource reference. ) collectionr[rQ)rrrREGISTRYrrZr[AsDict)rid_valuerrhrm parent_refs rGetResponseResourceRef3DeclarativeArgumentGenerator.GetResponseResourceRefsu!vhbG((<BB9MJ    # #$$..%%11  " $ $$rc[USS5$z.Gets the value of the limit flag (if present).limitNr8rs rLimit"DeclarativeArgumentGenerator.Limits 9gt ,,rc[USS5$z2Gets the value of the page size flag (if present). page_sizeNrrs rPageSize%DeclarativeArgumentGenerator.PageSizes 9k4 00r)rrl)NNNN) rrrrrrrrrrrrrrrrrrs=6!FCCP#'!%%) 5n$,-1rrcn\rSrSrSrSr1SkrSSjrSrSr Sr S r S r S r S rS rSrSSjrSrg)AutoArgumentGeneratoria3An argument generator to generate arguments for all fields in a message. When using this generator, you don't provide any manual configuration for arguments, it is all done automatically based on the request messages. There are two modes for this generator. In 'raw' mode, no modifications are done at all to the generated fields. In normal mode, certain list fields are not generated and instead our global list flags are used (and orchestrate the proper API fields automatically). In both cases, we generate additional resource arguments for path parameters. rz>filterorderBy pageTokencXlX lURRURR:gUl[ 5UlU(dURR5(aaU=R [R-slURR5nU(aUR RU5 gggg)aCreates a new Argument Generator. Args: method: APIMethod, The method to generate arguments for. raw: bool, True to do no special processing of arguments for list commands. If False, typical List command flags will be added in and the equivalent API fields will be ignored. N) rhrawdetailed_paramsrQ is_atomicr~ignored_fieldsHasTokenizedRequestrIGNORABLE_LIST_FIELDSBatchPageSizeFieldr)rrhrbatch_page_size_fields rrAutoArgumentGenerator.__init__sKH[[00DKK4F4FFDN%D 4;;2244 2HHH"kk<<>   56 53rc^^[5m/mUU4SjnU"UR55 U"URSURR 555 U"UR 55 T$)zGenerates all the CLI arguments required to call this method. Returns: {str, calliope.base.Action}, A map of field name to the argument. c>UH:nURnUT;aMTRU5 TRU5 M< g![a N"f=f)z Update args.N)rUr IndexErrorrk)rNrarUr+seens r _UpdateArgs7AutoArgumentGenerator.GenerateArgs.._UpdateArgssR# $ T\  ((4. C   s A AAr)r~_GenerateResourceArg_GenerateArgumentsrhr_GenerateListMethodFlags)rrr+rs @@rr"AutoArgumentGenerator.GenerateArgssa 5D D  ))+,''DKK,F,F,HIJ--/0 Krc HURR5nURUSU5nURU5nU(aQUR 5nUR URR Vs0sHnU[XFU5_M sn5 U"S0UD6$s snf)zGenerates the request object for the method call from the parsed args. Args: namespace: The argparse namespace. Returns: The apitools message to be send to the method. rr)rhr_ParseArguments_ParseResourceArgrr rQr8)rr request_typefieldsref relative_namefs rr#AutoArgumentGenerator.CreateRequest's;;--/L  ! !)R >F   +C &&(m mm"kk0020 66023  !& !!2s9Bc@UR(d [USS5$grrr8rs rrAutoArgumentGenerator.Limit<s 88 Y .. rc@UR(d [USS5$grr rs rrAutoArgumentGenerator.PageSizeCs 88 Y T 22 rc/nUR(dURR5(aUR[R 5 UR[R 5 URR5(a|URR5(a]UR[R5 URR5(aUR[R5 U$)z~Generates all the CLI flags for a List command. Returns: {str, calliope.base.Action}, A map of field name to the argument. ) rrhIsListrkr FILTER_FLAG SORT_BY_FLAGr ListItemField LIMIT_FLAGrPAGE_SIZE_FLAG)rrs rr.AutoArgumentGenerator._GenerateListMethodFlagsJs E 88 **,, ll4##$ ll4$$%  ( ( * *t{{/H/H/J/J T__% ;; ) ) + + ,,t** + Lrc./n[R"U5nUR5GHgnURURS5nUR URU5nU(dMEX-nUR [RR:XazURUS-UR5nU(aRU(aUS-U-OSn [R"U S9n URU 5 UHn U RU 5 M MM[ R""XwU5n [R$"X\SSS9n U R&RS 5(dS R)U5U R&S 'URU 5 GMj U$) aRGets the arguments to add to the parser that appear in the method body. Args: prefix: str, A string to prepend to the name of the flag. This is used for flags representing fields of a submessage. message: The apitools message to generate the flags for. Returns: {str, calliope.base.Argument}, A map of field name to argument. Nr7z: rrFMESSAGE) fix_boolscategoryrz"API doc needs help for field [{}].)r FieldHelpDocs all_fieldsgetrU _GetArgNamevariantrVariantrrr$rrMrk AddArgumentr rL GenerateFlagkwargsr:) rprefixr%r+ field_helpsfield field_helprUsub_args help_textgrouprarus rr(AutoArgumentGenerator._GenerateArguments\sM D))'2K##%??5::t4j   ejj* 5d  ]d (**22 2**4#:uzzB 4>td{Z/B)$$)4% ++e c   c " %--d*E $$U%.79zz~~f%%CJJ4P#**V  C)&* Krc </nURR(a URRROSnU(dU$[R"URR 55nSURR -n[R"[RSSS9nURU5 UHQn[R"SU-[R"U5SURXd5S9nURU5 MS U$) zGets the flags to add to the parser that appear in the method path. Returns: {str, calliope.base.Argument}, A map of field name to argument. NzFor substitution into: ?z+The GRI for the resource being operated on.)nargsrz--RESOURCE)metavarrr)rhrequest_collectionrr rr detailed_pathrrLrFLAT_RESOURCE_ARG_NAMErkrConvertToAngrySnakeCaser)rr+ field_namesr% default_helprar&s rr*AutoArgumentGenerator._GenerateResourceArgs Dkk44;;11AA:>  k))$++*D*D*FGK,t{{/H/HHL --44 : .s$r)rr@r[rQ)rhr1rr8rr3r DEFAULT_PARAMSrr]r rrrrZr[) rrr5rr@rQdefaultsrr=defaults rr'AutoArgumentGenerator._ParseResourceArgskk44;;11AA:>   0GGHA ,@$G FH i#e q **..q,?A 71+ VH K0@ @ OOF    # # kk44>>-KK22>> $ rNcU(a[R"U5(agXR;agXRR:Xa%UR 5R S5(agU$)aGets the name of the argument to generate for the field. Args: field_name: str, The name of the field. field_help: str, The help for the field in the API docs. Returns: str, The name of the argument to generate, or None if this field is output only or should be ignored. Nrequest)r IsOutputFieldrrh request_fieldlowerendswith)rr<r's rr!AutoArgumentGenerator._GetArgNames\i--j99 ((( kk///##I..  r)rrrhrFr")rrrrrr3rrrrrrrrrrrrrrrrrrsN &<7(>"*$"H>>&PrrrO),r __future__rrrrapitools.base.protorpclitergooglecloudsdk.callioper googlecloudsdk.calliope.conceptsrrs$googlecloudsdk.command_lib.util.apisr r r r r $googlecloudsdk.command_lib.util.argsrgooglecloudsdk.corergooglecloudsdk.core.resourcer Exceptionrrr'r.r5r=rHrPrdrqrwr|rrriobjectrrrrrrZs G&'/(B:7@DQ<):1I1OO@@ J .+\0F&*Z  M16M1`SFSr