;dZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z dd lmZd Zd Zd Ze j&j(de j&j*diZe j&j*fdZe j&j*dfdZdZdZdZ d"dZdZdZdZdZ dZ!dZ"dZ#dZ$dZ% d#d Z&y!)$zUtilities for the Config API.)absolute_import)division)unicode_literals)encoding)apis)waiter)base) resources)progress_trackerconfigi'iˤv1alpha2v1cDtjtt|S)zReturns the messages module for Config API. Args: release_track: The desired value of the enum googlecloudsdk.calliope.base.ReleaseTrack. Returns: Module containing the definitions of messages for Config API. )rGetMessagesModule _API_NAMERELEASE_TRACK_TO_API_VERSION) release_tracks >lib/googlecloudsdk/api_lib/infra_manager/configmanager_util.pyrr*s"   -m< TcJtjtt|| S)a$Returns an instance of the Config client. Args: release_track: The desired value of the enum googlecloudsdk.calliope.base.ReleaseTrack. use_http: bool, True to create an http object for this client. Returns: base_api.BaseApiClient, An instance of the Config client. )no_http)rGetClientInstancerr)ruse_https rrr9s'   "=1| rct}|j}|jj|j |S)aCalls into the GetDeployment API. Args: name: the fully qualified name of the deployment, e.g. "projects/p/locations/l/deployments/d". Returns: A messages.Deployment. Raises: HttpNotFoundError: if the deployment doesn't exist. name)rMESSAGES_MODULEprojects_locations_deploymentsGet,ConfigProjectsLocationsDeploymentsGetRequest)rclientmessagess r GetDeploymentr#KsA  &  # #(  . . 2 2;;;F rct}|j}|jj|j |||S)aCalls into the CreateDeployment API. Args: deployment: a messages.Deployment resource (containing properties like the blueprint). deployment_id: the ID of the deployment, e.g. "my-deployment" in "projects/p/locations/l/deployments/my-deployment". location: the location in which to create the deployment. Returns: A messages.OperationMetadata representing a long-running operation. )parent deployment deploymentId)rrrCreate/ConfigProjectsLocationsDeploymentsCreateRequest)r& deployment_idlocationr!r"s rCreateDeploymentr,_sJ  &  # #(  . . 5 5>>j}? rct}|j}|jj|j ||dS)a+Calls into the UpdateDeployment API. Args: deployment: a messages.Deployment resource (containing properties like the blueprint). deployment_full_name: the fully qualified name of the deployment. Returns: A messages.OperationMetadata representing a long-running operation. N)r&r updateMask)rrrPatch.ConfigProjectsLocationsDeploymentsPatchRequest)r&deployment_full_namer!r"s rUpdateDeploymentr2usK  &  # #(  . . 4 4==&:t> rc  t}tjj|jd}t j |j|j d _ fd} fd}dj|}tj|||}t j ||||tt}|S) a0Waits for the given "apply deployment" LRO to complete. Args: operation: the operation to poll. progress_message: string to display for default progress_tracker. Raises: apitools.base.py.HttpError: if the request returns an HTTP error. Returns: A messages.Deployment resource. $config.projects.locations.operations collectionctt}tj|j}tj|j |j }d}d}||j}|j}||dj|_ n||dj||_ |jya Updates the progress tracker with the result of the operation. Args: tracker: The ProgressTracker for the operation. result: the operation poll result. unused_status: map of stages with key as stage key (string) and value is the progress_tracker.Stage. r7Nz logs={0} zlogs={0}, step={1} ) rr MessageToJsonmetadata JsonToMessageOperationMetadatadeploymentMetadatalogsstepformatdetailed_messageTick) trackerresult unused_statusr"json_valdeployment_metadatar?r@pollers rTrackerUpdateFuncz:WaitForApplyDeploymentOperation..TrackerUpdateFuncs!"H%%foo6H"00""H D D& % %d % %d DL + 2 24 8f d. 5 < WaitForApplyDeploymentOperation..DetailMessageCallback  " ""r!Aborting wait for operation {0}. messagedetail_message_callbackaborted_messagecustom_trackertracker_update_func max_wait_mswait_ceiling_ms)rr REGISTRYParseRelativeNamerrCloudOperationPollerrprojects_locations_operationsrBrAr ProgressTrackerWaitFor_MAX_WAIT_TIME_MS_WAIT_CEILING_MS operationprogress_messager! operation_refrJrOrUrWrErIs @rWaitForApplyDeploymentOperationrgs  &$$66nn!G7-  & & ++ ** & &># 9?? N/#333%. >> #+#& & -rct}|j}|jj|j ||S)aCalls ImportStateFile API. Args: import_state_file_request: a messages.ImportStateFileRequest. deployment_id: the ID of the deployment, e.g. "my-deployment" in "projects/p/locations/l/deployments/my-deployment". Returns: (Statefile) The response message. )importStatefileRequestr%)rrr ImportState4ConfigProjectsLocationsDeploymentsImportStateRequest)import_state_file_requestr*r!r"s rImportStateFilermsI  &  # #(  . . : :CC!:=D rct}|j}|jj|j ||S)a_Calls ExportDeploymentStateFile API. Args: export_deployment_state_file_request: A ExportDeploymentStatefileRequest resource to be passed as the request body. deployment_id: the ID of the deployment, e.g. "my-deployment" in "projects/p/locations/l/deployments/my-deployment". Returns: (Statefile) The response message. ) exportDeploymentStatefileRequestr%)rrr ExportState4ConfigProjectsLocationsDeploymentsExportStateRequest)$export_deployment_state_file_requestr*r!r"s rExportDeploymentStateFilerssJ  &  # #(  . . : :CC+OD rct}|j}|jj|j ||S)adCalls ExportDeploymentRevisionsStateFile API. Args: export_revision_state_file_request: A ExportRevisionStatefileRequest resource to be passed as the request body. deployment_id: the ID of the deployment, e.g. "my-deployment" in "projects/p/locations/l/deployments/my-deployment". Returns: (Statefile) The response message. )exportRevisionStatefileRequestr%)rr(projects_locations_deployments_revisionsrp=ConfigProjectsLocationsDeploymentsRevisionsExportStateRequest)"export_revision_state_file_requestr*r!r"s rExportRevisionStateFilery sJ  &  # #(  8 8 D DLL)KM rct}|j}|jj|j |S)zCalls ExportLock API. Args: deployment_full_name: string, the fully qualified name of the deployment, e.g. "projects/p/locations/l/deployments/d". Returns: A lock info response. r)rrr ExportLock3ConfigProjectsLocationsDeploymentsExportLockRequestr1r!r"s rr{r{#sG  &  # #(  . . 9 9BB#C rct}|j}|jj|j ||S)a;Calls deployment Lock API. Args: lock_deployment_request: A LockDeploymentRequest resource to be passed as the request body deployment_full_name: the ID of the deployment, e.g. "my-deployment" in "projects/p/locations/l/deployments/my-deployment". Returns: A long running operation. )lockDeploymentRequestr)rrrLock-ConfigProjectsLocationsDeploymentsLockRequest)lock_deployment_requestr1r!r"s rLockDeploymentr7sI  &  # #(  . . 3 3<< 7#= rct}|j}|jj|j ||S)aHCalls deployment Unlock API. Args: unlock_deployment_request: A UnlockDeploymentRequest resource to be passed as the request body deployment_full_name: the ID of the deployment, e.g. "my-deployment" in format "projects/p/locations/l/deployments/my-deployment". Returns: A long running operation. )unlockDeploymentRequestr)rrrUnlock/ConfigProjectsLocationsDeploymentsUnlockRequest)unlock_deployment_requestr1r!r"s rUnlockDeploymentrQsI  &  # #(  . . 5 5>>";#? rct}|j}|jj|j |S)a+Lists all revisions for a deployment. Args: deployment_full_name: the fully qualified name of the deployment, e.g. "projects/p/locations/l/deployments/d". Returns: (ListRevisionsResponse) The response message. Raises: HttpNotFoundError: if the deployment doesn't exist. )r%)rrrvList6ConfigProjectsLocationsDeploymentsRevisionsListRequestr}s r ListRevisionsrksG  &  # #(  8 8 = =EE%F rct}|j}|jj|j ||S)a=Calls ExportPreviewResult API. Args: export_preview_result_request: A ExportPreviewResultRequest resource to be passed as the request body. preview_id: the ID of the preview, e.g. "my-preview" in "projects/p/locations/l/previews/my-preview". Returns: (Statefile) The response message. )exportPreviewResultRequestr%)rrprojects_locations_previewsExport,ConfigProjectsLocationsPreviewsExportRequest)export_preview_result_request preview_idr!r"s rExportPreviewResultrsI  &  # #(  + + 2 2;;%B< rct}|j}|jj|j |||S)aCalls into the CreatePreview API. Args: preview: a messages.Preview resource (containing properties like the blueprint). preview_id: the ID of the preview, e.g. "my-preview" in "projects/p/locations/l/previews/my-preview". location: the location in which to create the preview. Returns: A messages.OperationMetadata representing a long-running operation. )r%preview previewId)rrrr(,ConfigProjectsLocationsPreviewsCreateRequest)rrr+r!r"s r CreatePreviewrsJ  &  # #(  + + 2 2;;7j< rc  t}tjj|jd}t j |j|j d _ fd} fd}dj|}tj|||}t j ||||tt}|S) a+Waits for the given "create preview" LRO to complete. Args: operation: the operation to poll. progress_message: string to display for default progress_tracker. Raises: apitools.base.py.HttpError: if the request returns an HTTP error. Returns: A messages.Preview resource. r4r5r7ctt}tj|j}tj|j |j }d}d}||j}|j}||dj|_ n||dj||_ |jyr9) rrr:r;r<r=previewMetadatar?r@rArBrC) rDrErFr"rGpreview_metadatar?r@rIs rrJz8WaitForCreatePreviewOperation..TrackerUpdateFuncs!"H%%foo6H--""Ho D D#  " "d  " "d DL + 2 24 8f d. 5 < .DetailMessageCallbackrPrrQrRrV)rr r[r\rrr]rr^rBrAr r_r`rarbrcs @rWaitForCreatePreviewOperationrs  &$$66nn!G7-  & & (( ** & &># 9?? N/#333%. >> #+#& & -rN)zApplying the deployment)zCreating the preview)'__doc__ __future__rrrapitools.base.pyrgooglecloudsdk.api_lib.utilrrgooglecloudsdk.callioper googlecloudsdk.corer googlecloudsdk.core.consoler rrbra ReleaseTrackALPHAGArrrr#r,r2rgrmrsryr{rrrrrrrrrs$&'%,.()8  ' Z$  %)$5$5$8$8 %)$5$5$8$84$(,*!:Nb*44(44,4.!7Nr