2 dZddlmZddlmZddlmZddlmZddlZddlmZm Z ddl m Z dd l m Zdd lmZdd lm Zdd lmZdd lm Z dd lmZdd lmZddlm Z ddlmZddlmZddlmZdZdZdZdZdZ dZ!ejDdZ#dZ$dejJdede ejLfdZ'dejJdedejPfdZ) d3dejJdede eejTde ejVfdZ,dejJded ejLdejZfd!Z. d3dejJded ejLd"e e/dejZf d#Z0 d4dejJded$ejbd%e e2dejZf d&Z3dejJd'e/de ejbfd(Z4dejJd)ejZdejLfd*Z5dejJd)ejZdejbfd+Z6 d5dejJd)ejZfd,Z7d)ejZdejpfd-Z9dejJd.e/dejtfd/Z;d0e/d1etjttS)zReturns the messages module for the Runapps API. Returns: Module containing the definitions of messages for the Runapps API. )r GetMessagesModuleAPI_NAME API_VERSION8lib/googlecloudsdk/api_lib/run/integrations/api_utils.py GetMessagesr7s   + 66rclientapp_refreturnc"|jj|j} |jj |S#t j $rYyt j$r }t|j|Yd}~yd}~wwxYw)axCalls GetApplication API of Runapps for the specified reference. Args: client: The api client to use. app_ref: The resource reference of the application. Raises: exceptions.UnsupportedIntegrationsLocationError: if the region does not exist for the user. Returns: The application. If the application does not exist, then None is returned. nameN) MESSAGES_MODULE.RunappsProjectsLocationsApplicationsGetRequest RelativeNameprojects_locations_applicationsGetapitools_exceptionsHttpNotFoundErrorHttpForbiddenError_HandleLocationError locationsId)rrrequestes rGetApplicationr0@s$ KK##%L' 1  1 1 5 5g >>  . .   / /1,,a001sABB.B  Bc 4|jj|j}|jj |}|j rFt jdjdjt|j |S)zBCalls ListApplications API of Runapps for the specified reference.)parentzIThe following regions did not respond: {}. List results may be incompletez, ) r$/RunappsProjectsLocationsApplicationsListRequestr&r'r unreachabler warningformatjoinsorted)rrr.responses rListApplicationsr:]s  LL%%'M  3 3 8 8 A( KK ))/ IIfX112 3*  /r resource_idsc,|r*|Dcgc]}|jdz|jz }}ng}|j}|j|j |} |j j |Scc}w#tj$rYywxYw)awCalls GetApplicationStatus API of Runapps for the specified reference. Args: client: the api client to use. app_ref: the resource reference of the application. resource_ids: ResourceID of the resource to get status for. If not given, all resources in the application will be queried. Returns: The ApplicationStatus object. Or None if not found. /)r#rN) typer#r$4RunappsProjectsLocationsApplicationsGetStatusRequestr&r' GetStatusr)r*)rrr;res_id res_filtersmoduler.s rGetApplicationStatusrDss"6B6BF cFKK'lK  ! !&  G G    ![ H '  1 1 ; ;G DD  . . s#A8A==BB applicationc|jj|jj||j|j j S)aCalls CreateApplicaton API of Runapps for the specified reference. Args: client: the api client to use. app_ref: the resource reference of the application. application: the application to create Returns: the LRO of this request. )rE applicationIdr2)r'Creater$1RunappsProjectsLocationsApplicationsCreateRequestr#Parentr&)rrrEs rCreateApplicationrKsV  / / 6 6 NN!#((!..0O2 33r update_maskc|jj|jj|||j S)aLCalls ApplicationPatch API of Runapps for the specified reference. Args: client: the api client to use. app_ref: the resource reference of the application. application: the application to patch update_mask: comma separated string listing the fields to be updated. Returns: the LRO of this request. )rE updateMaskr#)r'Patchr$0RunappsProjectsLocationsApplicationsPatchRequestr&)rrrErLs rPatchApplicationrQsI"  / / 5 5 MM! ##%N' ((r deployment validate_onlyc|jj|jj|j ||j |S)a0Calls CreateDeployment API of Runapps. Args: client: the api client to use. app_ref: the resource reference of the application the deployment belongs to deployment: the deployment object validate_only: whether to only validate the deployment Returns: the LRO of this request. )r2rR deploymentId validateOnly)+projects_locations_applications_deploymentsrHr$/locations//applications//deployment/ Returns: the Deployment object. None is returned if the deployment cannot be found. r"N)rWr(r$9RunappsProjectsLocationsApplicationsDeploymentsGetRequestr)r*)rrZs r GetDeploymentr]sW  = = A A B B  C "   . . s47A  A  operationc0t|||jS)zWaits for an operation to complete. Args: client: client used to make requests. operation: object to wait for. Returns: the application from the operation. )_WaitForOperationr')rr^s rWaitForApplicationOperationras  69!AA CCrc4t|||j||S)a9Waits for an operation to complete. Args: client: client used to make requests. operation: object to wait for. tracker: The ProgressTracker that tracks the operation progress. tracker_update_func: function to update the tracker on polling. Returns: the deployment from thex operation. )r`rW)rr^trackertracker_update_funcs rWaitForDeploymentOperationres$ 69!MM"$7 99rc tj||j}tjj |j djt}fd} |jtj||tt|S#tj$rr|j|}tj dj|j"j$t'j(|j"j*t,j.$r,tj0djtdz wxYw)aWaits for an operation to complete. Args: client: client used to make requests. operation: object to wait for. resource_type: type, the expected type of resource response tracker: The ProgressTracker that tracks the operation progress. tracker_update_func: function to update the tracker on polling. Returns: The resulting resource of input paramater resource_type. z {}.projects.locations.operations) collectioncFyr ||yjyN)Tick)resultstatusrcrds r _StatusUpdatez(_WaitForOperation.._StatusUpdate3s# '662 llnr) max_wait_mswait_ceiling_ms status_updatez%OperationError: code={0}, message={1}zkOperation timed out after {0} seconds. The operations may still be underway remotely and may still succeed.r)r CloudOperationPollerprojects_locations_operationsrREGISTRYParseRelativeNamer#r6r GetResult PollUntilDone_POLLING_TIMEOUT_MS_RETRY_TIMEOUT_MSOperationErrorPollr IntegrationsOperationErrorerrorcoder Decodemessager WaitException TimeoutError)rr^ resource_typercrdpoller operation_refrms `` rr`r`s0"  & &}'-'K'K M&$$66nn3::8D7F--     +-'  ) **   M M*I  / //66 OO (//)//2I2I"J L MM   -    6 #d* + ---s (0BCEc|tj|jtj|jS)zGet the metadata message for the deployment operation. Args: messages: Module containing the definitions of messages for the Runapps API. operation: The LRO Returns: The DeploymentOperationMetadata object. )apitools_encodingPyValueToMessageDeploymentOperationMetadataMessageToPyValuemetadata)messagesr^s rGetDeploymentOperationMetadatarPs4  + +**((););< >>rproj_idc|jjdj|}|jj |S)zGet the list of all available regions from control plane. Args: client: instance of a client to use for the list request. proj_id: project id of the project to query. Returns: A list of location resources. z projects/{0}r")r$#RunappsProjectsLocationsListRequestr6projects_locationsr)rrr.s r ListLocationsrdsG  " " F F   ) G '  " " ' ' 00rregionr|ctj|}tj|jj r$t jdj||)zGet the metadata message for the deployment operation. Args: region: target region of the request. error: original HttpError. Raises: UnsupportedIntegrationsLocationError if it's location error. Otherwise raise the original error. z3Location {} is not found or access is unauthorized.) api_lib_exceptions HttpException_LOCATION_ERROR_REGEXmatchpayloadstatus_messager $UnsupportedIntegrationsLocationErrorr6)rr| parsed_errs rr,r,vsV"//6*  !3!3!B!BC  9 9=DDVL  +rri)F)NN)>__doc__ __future__rrrrretypingrrapitools.base.pyr rr r)googlecloudsdk.api_lib.utilr rr "googlecloudsdk.command_lib.runappsgooglecloudsdk.corer rgooglecloudsdk.core.utilr6googlecloudsdk.generated_clients.apis.runapps.v1alpha1rrrrAPP_DICT_CONFIG_KEYAPP_CONFIG_DICT_RESOURCES_KEYrwrxcompilerrRunappsV1alpha1 Applicationr0ListApplicationsResponser: ResourceIDApplicationStatusrD OperationrKstrrQ DeploymentboolrYr]rarer`rrListLocationsResponser Exceptionr,rrrrs)B&%' !:>,H.9#)-*Z\   +%" #BC71 # 3 31 1#$=$I$IJ1: # 3 3>G772JN # 3 3 4 9 D DEF'99: B3 # 3 33 3+663 # , , 36"& ( # 3 3( (+66(# (+D*M*M (8%*  # 3 3 *44D>  # , , 6 # 3 3%&?&J&JK.C # 3 3C(22C # . .C$9 # 3 39(229&?%I%I90#*. 1-5EE1-!:!D!D1-h>(22> # > >>(1 # 3 31 1.DD1$Y9r