o c@sdZddlmZddlmZddlmZddlZddlZddlZddlm Z ddl m Z ddl m Zdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm Z ddlm!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'ddl(Z(Gdddej)Z*Gdddej+Z,Gddde-Z.Gdddej/Z0d d!Z1d"d#Z2Gd$d%d%e(3ej4e-Z5Gd&d'd'e5Z6Gd(d)d)e5Z7Gd*d+d+e5Z8Gd,d-d-e5Z9Gd.d/d/e5Z:Gd0d1d1e5Z;Gd2d3d3e5ZGd8d9d9e5Z?Gd:d;d;e?Z@Gdd?d?e5ZBGd@dAdAe5ZCGdBdCdCe5ZDdS)DaAA yaml to calliope command translator. Calliope allows you to register a hook that converts a yaml command spec into a calliope command class. The Translator class in this module implements that interface and provides generators for a yaml command spec. The schema for the spec can be found in yaml_command_schema.yaml. )absolute_import)division)unicode_literalsN)messages)encoding) exceptions)HttpBadRequestError)waiter)base)command_loading) completers) arg_utils)registry)yaml_command_schema)yaml_command_schema_util) labels_util)log) resources) console_io)resource_transform)filesc@seZdZdZddZdS) Translatorz8Class that implements the calliope translator interface.cCst|d|}t||S)N)r CommandDataCommandBuilderGenerate)selfpath command_dataspecr Y/tmp/google-cloud-sdk/lib/googlecloudsdk/command_lib/util/apis/yaml_command_translator.py Translate<szTranslator.TranslateN)__name__ __module__ __qualname____doc__r"r r r r!r9s rcs0eZdZdZfddZddZddZZS)DeclarativeIamRolesCompleterzAn IAM role completer for a resource argument. The Complete() method override bypasses the completion cache. Attributes: _get_resource_ref: DeclarativeArgumentGenerator.GetPrimaryResource method to parse the resource ref. c s tt|jdi|||_dS)Nr )superr'__init___get_resource_ref)rget_resource_refkwargs __class__r r!r)Ks z%DeclarativeIamRolesCompleter.__init__cCs$||j}|}ddddd|gS)Niamzlist-grantable-rolesz--quietz--flatten=namez--format=disable)r* parsed_argsSelfLink)rparameter_info resource_ref resource_urir r r!GetListCommandOs z+DeclarativeIamRolesCompleter.GetListCommandcs,||}|||}fdd|pgDS)z;Bypasses the cache and returns completions matching prefix.cs"g|] }|dur|r|qSN) startswith).0itemprefixr r! [s z9DeclarativeIamRolesCompleter.Complete..)r5 GetAllItems)rr;r2commanditemsr r:r!CompleteWs   z%DeclarativeIamRolesCompleter.Complete)r#r$r%r&r)r5r@ __classcell__r r r-r!r'As  r'c@s0eZdZdZddZddZddZdd Zd S) rz3Generates calliope commands based on the yaml spec.cCsi|_|t|t|t|t|t|t|t|t |t |t |t |t |t|tdSr6)command_generatorsRegisterCommandGeneratorDescribeCommandGeneratorListCommandGeneratorDeleteCommandGeneratorCreateCommandGeneratorWaitCommandGeneratorUpdateCommandGeneratorGenericCommandGeneratorGetIamPolicyCommandGeneratorSetIamPolicyCommandGenerator#AddIamPolicyBindingCommandGenerator&RemoveIamPolicyBindingCommandGeneratorImportCommandGeneratorExportCommandGeneratorConfigExportCommandGeneratorrr r r!r)ds             zCommandBuilder.__init__cCs,|j|jvrtd|j||j|j<dS)Nz.Command type [{}] has already been registered.) command_typerB ValueErrorformat)rcommand_generatorr r r!rCws z'CommandBuilder.RegisterCommandGeneratorcCs4|j|jvrtdd||j|j|j|S)aReturns the command generator for a spec and path. Args: spec: yaml_command_schema.CommandData, the spec for the command being generated. path: Path for the command. Raises: ValueError: If we don't know how to generate the given command type (this is not actually possible right now due to the enum). Returns: The command generator. z'Command [{}] unknown command type [{}]. )rSrBrTrUjoin)rrrr r r!GetCommandGenerator}s  z"CommandBuilder.GetCommandGeneratorcCs|||}|S)aGenerates a calliope command from the yaml spec. Args: spec: yaml_command_schema.CommandData, the spec for the command being generated. path: Path for the command. Returns: calliope.base.Command, The command that implements the spec. )rYr)rrr generatorr r r!rs zCommandBuilder.GenerateN)r#r$r%r&r)rCrYrr r r r!ras  rc@sHeZdZdZddZeddZeddZdd Zd d Z d d Z dS)AsyncOperationPollerz(An implementation of a operation poller.cCs<||_||_|jjjsd|_n||_||_|o|jj|_dS)a4Creates the poller. Args: spec: yaml_command_schema.CommandData, the spec for the command being generated. resource_ref: resources.Resource, The resource reference for the resource being operated on (not the operation itself). If None, the operation will just be returned when it is done instead of getting the resulting resource. args: Namespace, The args namespace. operation_collection: str, collection name of operation method: registry.APIMethod, method used to make original api request N) rargsasync_extract_resource_resultr3_operation_collection collection full_name_resource_collection)rrr3r\operation_collectionmethodr r r!r)s zAsyncOperationPoller.__init__cCs,|jjjp |jjj}tj|j|jjj|dS)N api_version)rr]rfrequestr GetMethodr_rd)rrfr r r!operation_methods z%AsyncOperationPoller.operation_methodcCstj|jd|jjjdS)Ngetre)rrhrbrrgrfrRr r r!resource_get_methods z(AsyncOperationPoller.resource_get_methodcCst||jjjj}t|tjr|j}||jjjj vs"||jjjj vrBt||jjj j}|s7||jjjj vr7d}|r@t t|dSdS)z Overrides.zThe operation failed.TF)getattrrr]statefield isinstanceapitools_messagesEnumnamesuccess_values error_valueserrorr OperationErrorSerializeError)r operationresultrur r r!IsDones zAsyncOperationPoller.IsDonecCs|j}|}i}|D]}t||jjj|j |j |||j <q|di|}|jjj D] }|||j |}q0|j |S)zOverrides. Args: operation_ref: googlecloudsdk.core.resources.Resource. Returns: fetched operation message. Nr ) riGetRequestType RelativeName all_fieldsrlrr]operation_get_method_paramsrjrrmodify_request_hooksr\Call)r operation_ref request_type relative_namefieldsfrghookr r r!Polls    zAsyncOperationPoller.PollcCsH|}|jr|j}|}tj|j||dd||}t||jjj S)zOverrides. Args: operation: api_name_messages.Operation. Returns: result of result_service.Get request. T)is_primary_resource) r3rkr{r ParseResourceIntoMessager _GetAttributerr]result_attribute)rrxry get_methodrgr r r! GetResults    zAsyncOperationPoller.GetResultN) r#r$r%r&r)propertyrirkrzrrr r r r!r[s   r[cCsBt|tjr|Sz tjt|ddddWSty |YSw)z/Serializes the error message for better format.T),z: )indent sort_keys separators)rosix string_typesjsondumpsr MessageToDict Exception)rur r r!rws  rwc CsX|r*|dD]"}z|durWdSt||}Wqty)td|t|w|S)aXGets attributes and sub-attributes out of an object. Args: obj: The object to extract the attributes from. attr_path: str, The dotted path of attributes to extract. Raises: AttributeError: If the attribute doesn't exist on the object. Returns: The desired attribute or None if any of the parent attributes were None. .Nz*Attribute path [{}] not found on type [{}])splitrlAttributeErrorrUtype)obj attr_pathattrr r r!rs  rc@seZdZdZddZddZd,ddZd d Zd d Zd dZ ddZ d-ddZ d,ddZ ddZ ddZd,ddZddZ d.ddZd,d d!Zd"d#Zd$d%Zd&d'Zejd(d)Zd*d+ZdS)/BaseCommandGeneratorz"Base class for command generation.cCs"||_tj|j|_|dSr6)rr CommandTypeHasRequestMethodrShas_request_methodInitializeGeneratorForCommand)rrr r r!r)2s  zBaseCommandGenerator.__init__cCs:ddlm}|jr||_ng|_||jjj|_ dS)z*Initializes the arg_generator for command.rarg_marshallingN) $googlecloudsdk.command_lib.util.apisrr _GetMethodsmethodsDeclarativeArgumentGeneratorr argumentsparams arg_generator)rrr r r!r8s   z2BaseCommandGenerator.InitializeGeneratorForCommandNc CsFg}|jjjD]}|tj||p|jjj|jjj|jjjdq|S)N)disable_pagination) rrg collectionsappendrrhrdrfr)rrdrr`r r r!rFs  z BaseCommandGenerator._GetMethodscCs|j|j}||}|D]}||q|jjjr)|jjD]}||q!|jjj r7|j |jjj |jjj rG|j |jjj dSdS)zPerforms argument actions common to all commands. Adds all generated arguments to the parser Sets the command output format if specified Args: parser: The argparse parser. N)r GenerateArgsr_Exclude AddToParserrradditional_arguments_hookoutputrU display_info AddFormatflatten AddFlatten)rparserr\argr r r! _CommonArgsPs      z BaseCommandGenerator._CommonArgscs4fdd}tdddur|dSdS)zGenerates and registers a function to create a URI from a resource. Args: args: The argparse namespace. Returns: f(resource) -> str, A function that converts the given resource payload into a URI. cs:t|jjj}jjj}j||}| Sr6) rlrresponseid_fieldrGetPrimaryResourcerrdGetResponseResourceRefr1)resourceid_valuerdrefr\rr r!URIFuncos  z6BaseCommandGenerator._RegisterURIFunc..URIFuncuriNT)rlGetDisplayInfo AddUriFunc)rr\rr rr!_RegisterURIFunces  z%BaseCommandGenerator._RegisterURIFunccCs.|jjjD]}tjd|dd|q|S)zExcludes specified arguments from the parser. Args: parser: The argparse parser. Returns: The argparse parser. z--{}help)rrexcluder ArgumentrURemoveFromParser)rrrr r r!r{s zBaseCommandGenerator._ExcludecCsD|jjjs|jS|j|j|}||}|jj||}||Sr6)rrgmodify_method_hookrrrParser)rr\ specifiedrnew_method_namer r r!_GetRuntimeMethodss   z'BaseCommandGenerator._GetRuntimeMethodsc Cs:|||_|j|j|}|j}||}|jjjr7t j | |jjj|| || |||jjjddd|jjjrF||jj||fS|jjjrT|jj||}n5i}|r]|||jjjrj||jjj|jj||||jjj|jj|d}|jjjD]} | |||}q|j||j||j|d} || fS)a/Performs run actions common to all commands. Parses the resource argument into a resource reference Prompts the user to continue (if applicable) Calls the API method with the request generated from the parsed arguments Args: args: The argparse parser. existing_message: the apitools message returned from previous request. update_mask: auto-generated mask from updated fields. Returns: (resources.Resource, response), A tuple of the parsed resource reference and the API response from the method call. T)messagedefaultthrow_if_unattended cancel_on_noexisting_message)limit page_size)rrrrrdrrinputconfirmation_promptrPromptContinue_Format_GetDisplayResourceType_GetDisplayNamedefault_continuergissue_request_hookcreate_request_hookupdate static_fields CreateRequestrlabelsrSrrLimitPageSize) rr\r update_maskrrdrrgrrrr r r! _CommonRunsN           zBaseCommandGenerator._CommonRuncCst||||Sr6)rFormatResourceAttrStr)r format_stringr3 display_type display_namer r r!rszBaseCommandGenerator._FormatcCs8|j|j|j}|r|jr|||S|r|SdSr6)rrrprimary_resourcedisplay_name_hookName)rr3r\primary_resource_argr r r!rs  z$BaseCommandGenerator._GetDisplayNamecCs8|jjj}r |S|j|j|j}|r|js|jSdSr6) rrgdisplay_resource_typerrrris_parent_resourcerr)rr\ spec_displayrr r r!rs z,BaseCommandGenerator._GetDisplayResourceTypecCs|jjjrR|||jjjjd}|rRg}|jjjjr+|dt ||jjjj|jjjj r@|dt ||jjjj |rJt d |t t||jjjr_t ||jjj}|jjjD]}|||}qd|S)aAProcess the API response. Args: response: The apitools message object containing the API response. args: argparse.Namespace, The parsed args. Raises: core.exceptions.Error: If an error was detected and extracted from the response. Returns: A possibly modified response. rz Code: [{}]z Message: [{}]rW)rrru_FindPopulatedAttributernrcoderrUrrrErrorrXr text_typermodify_response_hooks)rrr\rurrr r r!_HandleResponses,     z$BaseCommandGenerator._HandleResponsec Cst|jjjD]B\}}ztjjt||jjj||jjj p |jj j d}||fWStj yI}z|t |jjjdkr?|WYd}~qd}~wwdS)N)r`rf) enumeraterr]rrREGISTRYrrlresponse_name_fieldrfrg UserErrorlen)rrxir`rer r r!_GetOperationRefs z%BaseCommandGenerator._GetOperationRefTc Cs||\}}|jjjp |}|r#tj|||||| |||jr:tj|d t j ||||S|j |j|j}t|j|rJ|nd|||} | |rZ| |S|j| ||dS)a6Handles polling for operations if the async flag is provided. Args: args: argparse.Namespace, The parsed args. resource_ref: resources.Resource, The resource reference for the resource being operated on (not the operation itself) operation: The operation message response. request_string: The format string to print indicating a request has been issued for the resource. If None, nothing is printed. extract_resource_result: bool, True to return the original resource as the result or False to just return the operation response when it is done. You would set this to False for things like Delete where the resource no longer exists when the operation is done. Returns: The response (either the operation or the original resource). z$Check operation [{{{}}}] for status.Nr\)rrr]request_issued_messagerstatusPrintrrrrUrREL_NAME_FORMAT_KEYrrrrdr[rzr_WaitForOperationWithPoller) rr\r3rxrequest_stringr^rrcrdpollerr r r! _HandleAsyncs6      z!BaseCommandGenerator._HandleAsyncc CsT|dtj|||}|r||j|nd}t|||||j|||S)Nz*Waiting for operation [{{{}}}] to complete) rrUrr rrr3r WaitFor)rrrr\progress_stringrr r r!r@s"  z0BaseCommandGenerator._WaitForOperationWithPollercCs||s|S|d}zt||}Wn tyYdSwt|tr4|D]}|||dd}|r3|Sq!|||ddS)aSearches the given object for an attribute that is non-None. This digs into the object search for the given attributes. If any attribute along the way is a list, it will search for sub-attributes in each item of that list. The first match is returned. Args: obj: The object to search attributes: [str], A sequence of attributes to use to dig into the resource. Returns: The first matching instance of the attribute that is non-None, or None if one could nto be found. rNr)rlrrolistr)rr attributesrxr r r!rMs  z,BaseCommandGenerator._FindPopulatedAttributec CsZddlm}|d}|j||}|j}|r|gng}||}|j}|| ||S)Nrrrj) rrrrrrrrdrr) rr\r get_methodsrrrget_arg_generatorrr r r!_GetExistingResourceks    z)BaseCommandGenerator._GetExistingResourcecs6|jjr t|}|jjrt|}|jjdur'|jjr"t|}nt|}|jj r4tj |jj |}|jj rDtj di|jj |}t |ds\dddfdd|jjD|_|jrtdd |jD}td d |jD}d |}d |}t|d krd |d|}nd|d|}||jd<|S)zConfigures top level attributes of the generated command. Args: command: The command being generated. Returns: calliope.base.Command, The command that implements the spec. N detailed_help DESCRIPTIONEXAMPLES) descriptionexamplescsi|] \}}|||qSr )rj)r8kvkey_mapr r! sz:BaseCommandGenerator._ConfigureCommand..css&|]}|jjd|jjVqdS)/N)r`api_namerfr8rdr r r! s  z9BaseCommandGenerator._ConfigureCommand..css|]}|jjVqdSr6)r`docs_urlr&r r r!r'sz, rzThis command uses *z?* APIs. The full documentation for these APIs can be found at: zThis command uses the *z<* API. The full documentation for this API can be found at: z API REFERENCEr )rhiddenr Hiddenauto_generated Autogenerateduniverse_compatibleUniverseCompatibleDefaultUniverseOnlyrelease_tracks ReleaseTracksdeprecated_data Deprecatehasattr help_textr?rrsetrrXr)rr> api_namesdoc_urls api_name_str doc_url_strapi_infor r!r!_ConfigureCommandzsJ            z&BaseCommandGenerator._ConfigureCommandcCsdSr6r rRr r r! _GenerateszBaseCommandGenerator._GeneratecCs|}|||Sr6)r=r<)rr>r r r!rs zBaseCommandGenerator.Generater6)NN)T)r#r$r%r&r)rrrrrrrrrrrrrrrrr<abcabstractmethodr=rr r r r!r/s2   =  " - 2  rc@s0eZdZdZejjZddZddZ ddZ dS) rJz&Generator for generic/custom commands.cCs|jjr tj|dSdSr6)rr]r ASYNC_FLAGr)rrr r r! _AddAsyncFlagsz%GenericCommandGenerator._AddAsyncFlagcCs|tdd|jD}|jj}|r|r|jrdStj|tj|tj |tj ||jj j r.)anyrrgenericdisable_paging_flagsr FILTER_FLAGr LIMIT_FLAGPAGE_SIZE_FLAG SORT_BY_FLAGrrURI_FLAG)rr is_paginatedrEr r r!_AddPagingFlagss     z'GenericCommandGenerator._AddPagingFlagscGfdddtj}|S)atGenerates a generic command. A generic command has a resource argument, additional fields, and calls an API method. It supports async if the async configuration is given. Any fields is message_params will be generated as arguments and inserted into the request message. Returns: calliope.base.Command, The command that implements the spec. c(eZdZefddZfddZdS)z2GenericCommandGenerator._Generate..Commandcs"|||dSr6)rrArMrrRr r!Argss  z7GenericCommandGenerator._Generate..Command.ArgscsR||\}}jjr#d}|rdtj}j||||d}||S)NRequest issued for: [{{{}}}]r) rrrr]rUrNAME_FORMAT_KEYrr)self_r\rrrrRr r!Runs  z6GenericCommandGenerator._Generate..Command.RunNr#r$r% staticmethodrQrVr rRr r!CommandrYr rYrrYr rRr!r=sz!GenericCommandGenerator._GenerateN) r#r$r%r&rrGENERICrSrArMr=r r r r!rJs  rJc@ eZdZdZejjZddZdS)rDz Generator for describe commands.crN)aGenerates a Describe command. A describe command has a single resource argument and an API method to call to get the resource. The result is returned using the default output format. Returns: calliope.base.Command, The command that implements the spec. crO)z3DescribeCommandGenerator._Generate..Commandc|dSr6rrPrRr r!rQ z8DescribeCommandGenerator._Generate..Command.Argscs|\}}||Sr6)rrrUr\ unused_refrrRr r!rVs z7DescribeCommandGenerator._Generate..Command.RunNrWr rRr r!rYsrY)r DescribeCommandr\r rRr!r=s z"DescribeCommandGenerator._GenerateN) r#r$r%r&rrDESCRIBErSr=r r r r!rD rDc@r^)rEzGenerator for list commands.crN)aGenerates a List command. A list command operates on a single resource and has flags for the parent collection of that resource. Because it extends the calliope base List command, it gets flags for things like limit, filter, and page size. A list command should register a table output format to display the result. If arguments.resource.response_id_field is specified, a --uri flag will also be enabled. Returns: calliope.base.Command, The command that implements the spec. crO)z/ListCommandGenerator._Generate..Commandcs(|jjjstj|dSdSr6)rrrrr rKrrPrRr r!rQ/s  z4ListCommandGenerator._Generate..Command.Argscs$||\}}||Sr6)rrrrbrRr r!rV7s  z3ListCommandGenerator._Generate..Command.RunNrWr rRr r!rY,rY)r ListCommandr\r rRr!r=szListCommandGenerator._GenerateN) r#r$r%r&rrLISTrSr=r r r r!rErfrEc@r^)rFzGenerator for delete commands.crN)aGenerates a Delete command. A delete command has a single resource argument and an API to call to perform the delete. If the async section is given in the spec, an --async flag is added and polling is automatically done on the response. For APIs that adhere to standards, no further configuration is necessary. If the API uses custom operations, you may need to provide extra configuration to describe how to poll the operation. Returns: calliope.base.Command, The command that implements the spec. crO)z1DeleteCommandGenerator._Generate..Commandcs&|jjrtj|dSdSr6)rrr]r r@rrPrRr r!rQYs z6DeleteCommandGenerator._Generate..Command.Argscsp|\}}jjr"j|||dtjdd}|jr"||S||}tj || |d|S)Nz#Delete request issued for: [{{{}}}]F)rr^kind) rrr]rrUrrTrrDeletedResourcerr)rUr\rrrRr r!rV_s"  z5DeleteCommandGenerator._Generate..Command.RunNrWr rRr r!rYVrZrY)r DeleteCommandr\r rRr!r=Dsz DeleteCommandGenerator._GenerateN) r#r$r%r&rrDELETErSr=r r r r!rF?rfrFc@r^)rGzGenerator for create commands.crN)aGenerates a Create command. A create command has a single resource argument and an API to call to perform the creation. If the async section is given in the spec, an --async flag is added and polling is automatically done on the response. For APIs that adhere to standards, no further configuration is necessary. If the API uses custom operations, you may need to provide extra configuration to describe how to poll the operation. Returns: calliope.base.Command, The command that implements the spec. crO)z1CreateCommandGenerator._Generate..Commandc:|jjrtj|jjjrt |dSdSr6) rrr]r r@rrrrAddCreateLabelsFlagsrPrRr r!rQ   z6CreateCommandGenerator._Generate..Command.Argsc s|\}}jj|j}|o|j}jjr:|dur&|s&dt j }nd}j ||||d}|jr: ||S|rWt |}|did}|sQ|d}t|} n||} tj| |d ||}|S)Nz#Create request issued for: [{{{}}}]zCreate request issuedrSrrrrj)rrrrrrrr]rUrrTrrrrrjrTransformBaseNamerrCreatedResourcer) rUr\rrrrr response_objra resource_namerRr r!rVsB       z5CreateCommandGenerator._Generate..Command.RunNrWr rRr r!rYrgrY)r CreateCommandr\r rRr!r=ys.z CreateCommandGenerator._GenerateN) r#r$r%r&rrCREATErSr=r r r r!rGtrfrGc@s,eZdZdZejjZ dddZddZ dS)rHzGenerator for wait commands.NcCs0t|j|r|nd||j|}|j|||dS)Nr )r[rGetCollectionInforar)rrr3r^rdr\rr r r!_WaitForOperations z&WaitCommandGenerator._WaitForOperationcrN)aPGenerates a wait command for polling operations. A wait command takes an operation reference and polls the status until it is finished or errors out. This follows the exact same spec as in other async commands except the primary operation (create, delete, etc) has already been done. For APIs that adhere to standards, no further async configuration is necessary. If the API uses custom operations, you may need to provide extra configuration to describe how to poll the operation. Returns: calliope.base.Command, The command that implements the spec. crO)z/WaitCommandGenerator._Generate..Commandcr_r6r`rPrRr r!rQraz4WaitCommandGenerator._Generate..Command.ArgscsDjj|}|j}||}j|dd||d}||}|S)NF)r3r^rdr\)rrrrdrryr)rUr\specified_resourcerdrrrRr r!rVs  z3WaitCommandGenerator._Generate..Command.RunNrWr rRr r!rYsrYr[r\r rRr!r=szWaitCommandGenerator._Generater6) r#r$r%r&rrWAITrSryr=r r r r!rHs  rHc@r^)rIzGenerator for update commands.cs(ddlmGfdddtj}|S)aGenerates an update command. An update command has a resource argument, additional fields, and calls an API method. It supports async if the async configuration is given. Any fields is message_params will be generated as arguments and inserted into the request message. Currently, the Update command is the same as Generic command. Returns: calliope.base.Command, The command that implements the spec. r)rcs*eZdZefddZfddZdS)z1UpdateCommandGenerator._Generate..Commandcror6) rrr]r r@rrrrAddUpdateLabelsFlagsrPrRr r!rQrqz6UpdateCommandGenerator._Generate..Command.Argsc sd}jjrjjjr|}|_jj|j} |}|r?jjr2jjj r2d}n |j|}||i}nd} |||\}}jj rad} |rXdtj} j|||| d}tj|||d||S)NrrRrSrj)rrread_modify_updaterrrrrrdGetMaskFieldPathdisable_auto_field_mask GetMaskStringrr]rUrrTrrUpdatedResourcerrr) rUr\rrd mask_path mask_stringrrrrrrr r!rVsD       z5UpdateCommandGenerator._Generate..Command.RunNrWr rr r!rY srY)rrr rYr\r rr!r=s 2z UpdateCommandGenerator._GenerateN) r#r$r%r&rrUPDATErSr=r r r r!rIrfrIc@r^)rKz&Generator for get-iam-policy commands.cs&ddlm}Gfdddtj}|S)a%Generates a get-iam-policy command. A get-iam-policy command has a single resource argument and an API method to call to get the resource. The result is returned using the default output format. Returns: calliope.base.Command, The command that implements the spec. riam_utilcs,eZdZdZefddZfddZdS)z7GetIamPolicyCommandGenerator._Generate..CommandzGet IAM policy command closure.cs|tj|dSr6)rr rKrrPrRr r!rQZs z.Command.ArgscsFjjrjjjrjjjjjjjjj<|\}}||Sr6)rr/policy_versionrgrget_iam_policy_version_pathrr)rUr\_rrRr r!rV_s z;GetIamPolicyCommandGenerator._Generate..Command.RunNr#r$r%r&rXrQrVr rRr r!rYWs rY)googlecloudsdk.command_lib.iamrr rh)rrrYr rRr!r=Es z&GetIamPolicyCommandGenerator._GenerateN) r#r$r%r&rrGET_IAM_POLICYrSr=r r r r!rK@rfrKc@s(eZdZdZejjZddZddZ dS)rLz&Generator for set-iam-policy commands.cCspd}d}|jjr|jjj}d|vr|dp|}|jjjp|}d|}||}t|dr6||jjj|<dSdS)aQSet Field Mask on SetIamPolicy request message. If the API supports update_masks then adds the update_mask to the SetIamPolicy request (via static fields). Args: update_mask: str, comma separated string listing the Policy fields to be updated. method: APIMethod, used to identify update mask field. SetIamPolicyRequestsetIamPolicyRequestset_iam_policy_requestz {}.updateMask updateMaskN) rr/message_type_overridesset_iam_policy_request_pathrUGetMessageByNamer4rgr)rrrdrpolicy_request_path overridesmask_field_pathupdate_requestr r r!_SetPolicyUpdateMaskps      z1SetIamPolicyCommandGenerator._SetPolicyUpdateMaskc(ddlmGfdddtj}|S)aQGenerates a set-iam-policy command. A set-iam-policy command takes a resource argument, a policy to set on that resource, and an API method to call to set the policy on the resource. The result is returned using the default output format. Returns: calliope.base.Command, The command that implements the spec. rrc0eZdZdZefddZfddZdS)z7SetIamPolicyCommandGenerator._Generate..CommandzSet IAM policy command closure.cs$||tj|dSr6)rAddArgForPolicyFiler rKrrPrrr r!rQs  z.Command.Argsc sd}d}jjrdjjjvrjjjdp|}jjjp|}|d}jj|j}||}|s:t d | |j |\}}jjrRjjj rRjjj |_|jjj|<||z |\} } Wnty|} ztjd| d} ~ ww| || |S) Called when command is executed.Policyrpolicy.policyzPolicy type [{}] not found.zERROR: Policy modification failed. For bindings with conditions, run "gcloud alpha iam policies lint-condition" to identify issues in conditions.N)rr/rrrrrrdrrTrUParsePolicyFileWithUpdateMask policy_filerversionrgrrrrrerrr LogSetIamPolicyrrr) rUr\policy_type_namerpolicy_field_pathrd policy_typerrrrexrr r!rVsR     z;SetIamPolicyCommandGenerator._Generate..Command.RunNrr rr r!rYs rYrrr rYr\r rr!r=s 6z&SetIamPolicyCommandGenerator._GenerateN) r#r$r%r&rrSET_IAM_POLICYrSrr=r r r r!rLks  rLc@s@eZdZdZeddZeddZddZdd Zd d Z d S) $BaseIamPolicyBindingCommandGeneratorz.Base class for iam binding command generators.cC|jjo|jjjSr6)rr/enable_conditionrRr r r!_add_conditionz3BaseIamPolicyBindingCommandGenerator._add_conditioncCrr6)rr/hide_special_member_typesrRr r r!_hide_special_member_typesrz?BaseIamPolicyBindingCommandGenerator._hide_special_member_typescCs||}|j|||Sr6)rrrr)rr\rr r r!_GetResourceRefs z4BaseIamPolicyBindingCommandGenerator._GetResourceRefcs |jGfdddtS)zGenerate a IAM role completer.cs eZdZfddZZS)z]BaseIamPolicyBindingCommandGenerator._GenerateDeclarativeIamRolesCompleter..Completercst|jddi|dS)Nr+r )r(r))rr,) Completerr.r+r r!r)s  zfBaseIamPolicyBindingCommandGenerator._GenerateDeclarativeIamRolesCompleter..Completer.__init__)r#r$r%r)rAr rr+r-r!rsr)rr'rRr rr!%_GenerateDeclarativeIamRolesCompleterszJBaseIamPolicyBindingCommandGenerator._GenerateDeclarativeIamRolesCompletercCsb|d}|j||j}|j||}|jjr*|jjjr*t ||jjj |jjj| |}|S)z4GetIamPolicy helper function for add/remove binding. getIamPolicy) rrrrdrrr/rr SetFieldInMessagerr)rr\get_iam_methodsget_iam_methodget_iam_requestrr r r! _GetIamPolicys"  z2BaseIamPolicyBindingCommandGenerator._GetIamPolicyN) r#r$r%r&rrrrrrr r r r!rs   rc@*eZdZdZejjZdddZddZ dS) rMz.Generator for add-iam-policy binding commands.Fc Csddlm}|j|j|j}|d}|r3||}||}|d}| ||||j |j ||S||}| |||j |j |S)zGet the IAM policy and add the specified binding to it. Args: args: an argparse namespace. add_condition: True if support condition. Returns: IAM policy. rrBindingExpr) rrrrrrdr$ValidateAndExtractConditionMutexRoler"AddBindingToIamPolicyWithConditionmemberroleAddBindingToIamPolicy) rr\ add_conditionrrdbinding_message_type conditionrcondition_message_typer r r!"_GetModifiedIamPolicyAddIamBindings"       zFAddIamPolicyBindingCommandGenerator._GetModifiedIamPolicyAddIamBindingcr)aGenerates an add-iam-policy-binding command. An add-iam-policy-binding command adds a binding to a IAM policy. A binding consists of a member, a role to define the role of the member, and an optional condition to define in what condition the binding is valid. Two API methods are called to get and set the policy on the resource. Returns: calliope.base.Command, The command that implements the spec. rrcr)z>AddIamPolicyBindingCommandGenerator._Generate..Commandz'Add IAM policy binding command closure.c4j|jjd|tj|dSN)role_completerrr)AddArgsForAddIamPolicyBindingrrrrr rKrrPrr r!rQA zCAddIamPolicyBindingCommandGenerator._Generate..Command.Argsc sd}jjr jjjp |}|d}j|jd}jjr(jjjr(jjj|_|jjj|<z  |\}}Wnt yL}zt j d|d}~ww||||S)rrrrzERROR: Policy modification failed. For a binding with condition, run "gcloud alpha iam policies lint-condition" to identify issues in condition.N)rr/rrrrrrgrrrrrr rrrr)rUr\rrrrrrrr r!rVKs,   zBAddIamPolicyBindingCommandGenerator._Generate..Command.RunNrr rr r!rY>  rYrr\r rr!r=+s -z-AddIamPolicyBindingCommandGenerator._GenerateNF) r#r$r%r&rrADD_IAM_POLICY_BINDINGrSrr=r r r r!rM s   rMc@r) rNz1Generator for remove-iam-policy binding commands.FcCsbddlm}|r!||}||}|j||j|j||jd|S||}|||j|j|S)zGet the IAM policy and remove the specified binding to it. Args: args: an argparse namespace. add_condition: True if support condition. Returns: IAM policy. rr)all_conditions) rrValidateAndExtractConditionr'RemoveBindingFromIamPolicyWithConditionrrallRemoveBindingFromIamPolicy)rr\rrrrr r r!%_GetModifiedIamPolicyRemoveIamBindingts    zLRemoveIamPolicyBindingCommandGenerator._GetModifiedIamPolicyRemoveIamBindingcr)aGenerates a remove-iam-policy-binding command. A remove-iam-policy-binding command removes a binding from a IAM policy. A binding consists of a member, a role to define the role of the member, and an optional condition to define in what condition the binding is valid. Two API methods are called to get and set the policy on the resource. Returns: calliope.base.Command, The command that implements the spec. rrcr)zARemoveIamPolicyBindingCommandGenerator._Generate..Commandz*Remove IAM policy binding command closure.crr) AddArgsForRemoveIamPolicyBindingrrrrr rKrrPrr r!rQrzFRemoveIamPolicyBindingCommandGenerator._Generate..Command.Argscsd}jjr jjjp |}|d}j|jd}jjr(jjjr(jjj|_|jjj|< |\}} |  | ||S)rrrr)rr/rrrrrrgrrrrrr)rUr\rrrrrrr r!rVs   zERemoveIamPolicyBindingCommandGenerator._Generate..Command.RunNrr rr r!rYrrYrr\r rr!r=s $z0RemoveIamPolicyBindingCommandGenerator._GenerateNr) r#r$r%r&rrREMOVE_IAM_POLICY_BINDINGrSrr=r r r r!rNns   rNc@r^)rOzGenerator for import commands.cr)aLGenerates an import command. An import command has a single resource argument and an API method to call to get the resource. The result is from a local yaml file provided by the `--source` flag, or from stdout if nothing is provided. Returns: calliope.base.Command, The command that implements the spec. rutilc.eZdZdZefddZfddZdS)z1ImportCommandGenerator._Generate..CommandzImport command enclosure.cs0|jjrtj||jddddS)Nz--sourcea Path to a YAML file containing the configuration export data. The YAML file must not contain any output-only fields. Alternatively, you may omit this flag to read from standard input. For a schema describing the export/import format, see: $CLOUDSDKROOT/lib/googlecloudsdk/schemas/... $CLOUDSDKROOT is can be obtained with the following command: $ gcloud info --format='value(installation.sdk_root)' r)rrr]r r@r add_argumentrPrRr r!rQs   z6ImportCommandGenerator._Generate..Command.Argsc sjj|j}|}|j}||j}tj |j pddd} |j j jjj|j}j|||d}d} jjrjjj} jjj} z|} Wn=tjy} z0| jdks\| s^| jjjj_jjjrodj_n jjjr{jjjj_WYd} ~ nd} ~ ww| r|| krtj !d"|j#Sj$||d\} }jjrd}| durd"t%j&}'|| ||}(||S) N-F)binary) message_typestream schema_pathiz/Request not sent for [{}]: No changes detected.rrR))rrrrdr{ request_field field_by_namerrReadFromFileOrStdinsource GetSchemaPathr`r%rrgrfr#Importimport_abort_if_equivalentcreate_if_not_existsrapitools_exceptions HttpError status_codecreate_requestno_create_asyncr] create_asyncrrr r rUrrrrrTrr)rUr\rdrrresource_message_classdatarimported_resourceexisting_resourcerrrurrr export_utilrr r!rVsn         z5ImportCommandGenerator._Generate..Command.RunNrr rr r!rYs rY)!googlecloudsdk.command_lib.exportrr ImportCommandr\r rr!r=s Xz ImportCommandGenerator._GenerateN) r#r$r%r&rrIMPORTrSr=r r r r!rOrfrOc@r^)rPzGenerator for export commands.cr)aVGenerates an export command. An export command has a single resource argument and an API method to call to get the resource. The result is exported to a local yaml file provided by the `--destination` flag, or to stdout if nothing is provided. Returns: calliope.base.Command, The command that implements the spec. rrcr)z1ExportCommandGenerator._Generate..CommandzExport command enclosure.cs||jddddS)Nz --destinationaf Path to a YAML file where the configuration will be exported. The exported data will not contain any output-only fields. Alternatively, you may omit this flag to write to standard output. For a schema describing the export/import format, see $CLOUDSDKROOT/lib/googlecloudsdk/schemas/... r)rrrPrRr r!rQPs  z6ExportCommandGenerator._Generate..Command.Argscs|\}}jj|j}|jjjj j t |j }| drNt|j}j|||dWdn1s=wYtjd|j|jSj|tj|ddS)N destination)rrrzExported [{}] to '{}'.)rrrrrdrr`r%rrgrfrr# IsSpecifiedr FileWriterrExportrr r rUrrsysstdout)rUr\rcrrdrrrr r!rV]s,    z5ExportCommandGenerator._Generate..Command.RunNrr rr r!rYMs  rY)rrr ExportCommandr\r rr!r=;s #z ExportCommandGenerator._GenerateN) r#r$r%r&rrEXPORTrSr=r r r r!rP6rfrPc@r^)rQz%Generator for config export commands.cs6ddlmddlmGfdddtj}|S)aGenerates a config export command. A config export command has a resource argument as well as configuration export flags (such as --output-format and --path). It will export the configuration for one resource to stdout or to file, or will output a stream of configurations for all resources of the same type within a project to stdout, or to multiple files. Supported formats are `KRM` and `Terraform`. Returns: calliope.base.Command, The command that implements the spec. r)flags)python_command_utilcs,eZdZefddZfddZdS)z7ConfigExportCommandGenerator._Generate..Commandcsx|jddd}|}jj}|D]}|jD]\}}d|_q||qj|dd | |dS)NT)mutexrequiredFproject)r`) add_grouprrrspecsr?rr AddAllFlag AddPathFlag AddFormatFlag)r mutex_groupresource_groupr\rrvalue)declarative_config_flagsrr r!rQs  z.Command.ArgscsVjj|j}|o |j}t|ddrj||jddSj||j|| dS)Nr)r\r`r3) rrrrr`rl RunExportraParseResourceArgr1)rUr\ resource_argr`)rrr r!rVs   z;ConfigExportCommandGenerator._Generate..Command.RunNrWr rrrr r!rYs rY)+googlecloudsdk.command_lib.util.declarativerrr rYr\r rr!r=xs z&ConfigExportCommandGenerator._GenerateN) r#r$r%r&rr CONFIG_EXPORTrSr=r r r r!rQsrfrQ)Er& __future__rrrr>rrapitools.base.protorpcliterrpapitools.base.pyrrrapitools.base.py.exceptionsrgooglecloudsdk.api_lib.utilr googlecloudsdk.callioper r googlecloudsdk.command_lib.utilr rr rrr$googlecloudsdk.command_lib.util.argsrgooglecloudsdk.corerrgooglecloudsdk.core.consolergooglecloudsdk.core.resourcergooglecloudsdk.core.utilrrYamlCommandTranslatorrListCommandCompleterr'objectrOperationPollerr[rwrwith_metaclassABCMetarrJrDrErFrGrHrIrKrLrrMrNrOrPrQr r r r!sl                       ?f ? *5H5O+p. eWq=