dZddlmZddlmZddlmZddlmZmZddlm Z ddlm Z ddl m Zdd lmZddlm Z dd lmZdd lZGd d eZy )z%Cloud Functions (2nd gen) API Client.)absolute_import)division)unicode_literals) GeneratorOptional) exceptions) list_pager)util)types) propertiesNc :eZdZdZdZdeejddffdZdde de e fdZ e j dd e d ede ejfd Ze jd e dej"fd Ze jd e dej"fd Ze jd e dej"fdZe jd e dej"fdZe jd e de dej"fdZe jd e dej"fdZy)FunctionsClientz)Client for Cloud Functions (2nd gen) API.cltj||_tj||_yN)r GetClientInstanceclientGetMessagesModulemessages)self release_tracks 1lib/googlecloudsdk/api_lib/functions/v2/client.py__init__zFunctionsClient.__init__#s&((7DK**=9DMreturnNctjjjj }|j j d|z}tj|jj|ddS)zLists GCF gen2 regions. Returns: Iterable[cloudfunctions_v2alpha.Location], Generator of available GCF gen2 regions. z projects/name locationspageSize)servicerequestfieldbatch_size_attribute) r VALUEScoreproject GetOrFailr*CloudfunctionsProjectsLocationsListRequestr YieldFromListrprojects_locations)rr&r!s r ListRegionszFunctionsClient.ListRegions'sq$$,,668GmmFF 7 "GG  # # ..'  rregion query_filterc tjjjj }|j j dj|||}|jjj|S)aLists available GCF Gen 2 Runtimes in a region. Args: region: str, The region targeted to list runtimes in. query_filter: str, Filters to apply to the list runtimes request. Returns: v2alpha|v2beta.ListRuntimesResponse, The list runtimes request z%projects/{project}/locations/{region})r&r,)parentfilter) r r$r%r&r'r2CloudfunctionsProjectsLocationsRuntimesListRequestformatrprojects_locations_runtimesList)rr,r-r&r!s r ListRuntimeszFunctionsClient.ListRuntimes9sz$$,,668GmmNN6==F>  OG ;; 2 2 7 7 @@rrraise_if_not_foundc* |jjj|jj |S#t j $r>}|s1|jtjjjk7rYd}~yd}~wwxYw)a=Gets the function with the given name or None if not found. Args: name: GCFv2 function resource relative name. raise_if_not_found: If set, raises NOT_FOUND http errors instead of returning None. Returns: cloudfunctions_v2_messages.Function, the fetched GCFv2 function or None. rN) rprojects_locations_functionsGetr2CloudfunctionsProjectsLocationsFunctionsGetRequestapitools_exceptions HttpError status_codesixmoves http_client NOT_FOUND)rrr6errors r GetFunctionzFunctionsClient.GetFunctionOs~  [[ 5 5 9 9 -- J J K   ( (    syy44>> >  s>AB4B  Bc|jjj|jj |S)zAborts the function upgrade for the given function. Args: name: str, GCFv2 function resource relative name. Returns: A long-running operation. r)rr8AbortFunctionUpgraderCCloudfunctionsProjectsLocationsFunctionsAbortFunctionUpgradeRequestrrs rrEz$FunctionsClient.AbortFunctionUpgradeks< ;; 3 3 H H YY Z  rc|jjj|jj |S)zCommits the function upgrade for the given function. Args: name: str, GCFv2 function resource relative name. Returns: A long-running operation. r)rr8CommitFunctionUpgraderDCloudfunctionsProjectsLocationsFunctionsCommitFunctionUpgradeRequestrGs rrIz%FunctionsClient.CommitFunctionUpgrade{s< ;; 3 3 I I ZZ [  rc|jjj|jj |S)zRedirects function upgrade traffic for the given function. Args: name: str, GCFv2 function resource relative name. Returns: A long-running operation. r)rr8RedirectFunctionUpgradeTrafficrMCloudfunctionsProjectsLocationsFunctionsRedirectFunctionUpgradeTrafficRequestrGs rrLz.FunctionsClient.RedirectFunctionUpgradeTraffic< ;; 3 3 R R cc d  rc|jjj|jj |S)zRolls back function upgrade traffic for the given function. Args: name: str, GCFv2 function resource relative name. Returns: A long-running operation. r)rr8RollbackFunctionUpgradeTrafficrMCloudfunctionsProjectsLocationsFunctionsRollbackFunctionUpgradeTrafficRequestrGs rrPz.FunctionsClient.RollbackFunctionUpgradeTrafficrNrtrigger_service_accountc|jjj|jj ||jj |S)aSets up the function upgrade config for the given function. Args: name: str, GCFv2 function resource relative name. trigger_service_account: str, The service account to use for the trigger. Returns: A long-running operation. )triggerServiceAccount)r!setupFunctionUpgradeConfigRequest)rr8SetupFunctionUpgradeConfigrICloudfunctionsProjectsLocationsFunctionsSetupFunctionUpgradeConfigRequest!SetupFunctionUpgradeConfigRequest)rrrRs rrVz*FunctionsClient.SetupFunctionUpgradeConfigsW ;; 3 3 N N __.2mm.].]&=/^/ `  rc|jjj|jj |S)a&Detach a GCF 2nd gen function from GCF and make it a native Cloud Run function. Args: name: str, GCFv2 function resource relative name which follows the format of `projects/{project}/locations/{region}/functions/{function}`. Returns: A long-running operation. r)rr8DetachFunctionr=CloudfunctionsProjectsLocationsFunctionsDetachFunctionRequestrGs rrZzFunctionsClient.DetachFunctions< ;; 3 3 B B SS T  rr)F)__name__ __module__ __qualname____doc__rrr Locationr+strrr5util_v1 CatchHTTPErrorRaiseHTTPExceptionboolFunctionrC OperationrErIrLrPrVrZrrrr s1:9U^^T4%?@$AAHSMA, ++27+/,6 ++ s u ,  ++   ,  ++   ,  ++   ,  ++03 ,* ++,rr)r_ __future__rrrtypingrrapitools.base.pyrr;r #googlecloudsdk.api_lib.functions.v1r rb#googlecloudsdk.api_lib.functions.v2r googlecloudsdk.corer r>objectrrgrrros9,&'&>'?54* pfpr