
    )W                     V    S r SSKJr  SSKJr  SSKJr   " S S\R                  5      r	g)z:Generated client library for bigquerymigration version v2.    )absolute_import)base_api)bigquerymigration_v2_messagesc                   h  ^  \ rS rSrSr\rSrSrSr	S/r
SrSrS	rS
rS rSrSr     SU 4S jjr " S S\R(                  5      r " S S\R(                  5      r " S S\R(                  5      r " S S\R(                  5      r " S S\R(                  5      r " S S\R(                  5      rSrU =r$ )BigquerymigrationV2
   zBGenerated client library for service bigquerymigration version v2.z)https://bigquerymigration.googleapis.com/z.https://bigquerymigration.mtls.googleapis.com/bigquerymigrationz.https://www.googleapis.com/auth/cloud-platformv2	CLIENT_IDCLIENT_SECRETzgoogle-cloud-sdkNc                 h  > U=(       d    U R                   n[        [        U ]  XX4UXgUU	U
US9  U R	                  U 5      U l        U R                  U 5      U l        U R                  U 5      U l	        U R                  U 5      U l        U R                  U 5      U l        U R                  U 5      U l        g)z&Create a new bigquerymigration handle.)
credentialsget_credentialshttpmodellog_requestlog_responsecredentials_argsdefault_global_paramsadditional_http_headersresponse_encodingN)BASE_URLsuperr   __init__$ProjectsLocationsSubtaskTypesServiceprojects_locations_subtaskTypes!ProjectsLocationsTaskTypesServiceprojects_locations_taskTypes)ProjectsLocationsWorkflowsSubtasksService%projects_locations_workflows_subtasks!ProjectsLocationsWorkflowsServiceprojects_locations_workflowsProjectsLocationsServiceprojects_locationsProjectsServiceprojects)selfurlr   r   r   r   r   r   r   r   r   r   	__class__s               ]lib/googlecloudsdk/generated_clients/apis/bigquerymigration/v2/bigquerymigration_v2_client.pyr   BigquerymigrationV2.__init__   s     
C	
t-'%)3 7+ . - ,0+T+TUY+ZD((,(N(Nt(TD%151_1_`d1eD.(,(N(Nt(TD%";;DAD((.DM    c                   ~   ^  \ rS rSrSrSrU 4S jrSS jrS \l        SS jr	S \	l        SS	 jr
S
 \
l        SrU =r$ )8BigquerymigrationV2.ProjectsLocationsSubtaskTypesService1   z?Service class for the projects_locations_subtaskTypes resource.r   c                 N   > [         [        R                  U ]  U5        0 U l        g N)r   r   r   r   _upload_configsr'   clientr)   s     r*   r   ABigquerymigrationV2.ProjectsLocationsSubtaskTypesService.__init__6   s$    DDdTU[\dr,   c                 D    U R                  S5      nU R                  X1US9$ )a  Finishes a migration subtask via the subtask type.

Args:
  request: (BigquerymigrationProjectsLocationsSubtaskTypesFinishSubtaskRequest) input message
  global_params: (StandardQueryParameters, default: None) global arguments
Returns:
  (GoogleProtobufEmpty) The response message.
FinishSubtaskglobal_paramsGetMethodConfig
_RunMethodr'   requestr9   configs       r*   r7   FBigquerymigrationV2.ProjectsLocationsSubtaskTypesService.FinishSubtask;   s/     ##O4f__
  8 8r,   c                  B    [         R                  " SSSS/S// SSSSS	S
9$ )Nz\v2/projects/{projectsId}/locations/{locationsId}/subtaskTypes/{subtaskTypesId}:finishSubtaskPOSTz?bigquerymigration.projects.locations.subtaskTypes.finishSubtasksubtaskTypezv2/{+subtaskType}:finishSubtask2googleCloudBigqueryMigrationV2FinishSubtaskRequestBBigquerymigrationProjectsLocationsSubtaskTypesFinishSubtaskRequestGoogleProtobufEmptyF	flat_pathhttp_method	method_idordered_paramspath_paramsquery_paramsrelative_pathrequest_fieldrequest_type_nameresponse_type_namesupports_downloadr   ApiMethodInfo r,   r*   <lambda>ABigquerymigrationV2.ProjectsLocationsSubtaskTypesService.<lambda>H   s4    (*@*@pS%"O7J^0+r,   c                 D    U R                  S5      nU R                  X1US9$ )a  Assigns a migration subtask to the caller, if available. Example: `/v2/projects/123/locations/us/subtaskTypes:receiveMigrationSubtask`.

Args:
  request: (BigquerymigrationProjectsLocationsSubtaskTypesReceiveMigrationSubtaskRequest) input message
  global_params: (StandardQueryParameters, default: None) global arguments
Returns:
  (GoogleCloudBigqueryMigrationV2ReceiveMigrationSubtaskResponse) The response message.
ReceiveMigrationSubtaskr8   r:   r=   s       r*   rY   PBigquerymigrationV2.ProjectsLocationsSubtaskTypesService.ReceiveMigrationSubtaskV   s0     ##$=>f__
  8 8r,   c                  B    [         R                  " SSSS/S// SSSSS	S
9$ )NzUv2/projects/{projectsId}/locations/{locationsId}/subtaskTypes:receiveMigrationSubtaskrB   zIbigquerymigration.projects.locations.subtaskTypes.receiveMigrationSubtaskprojectLocationz-v2/{+projectLocation}:receiveMigrationSubtask<googleCloudBigqueryMigrationV2ReceiveMigrationSubtaskRequestLBigquerymigrationProjectsLocationsSubtaskTypesReceiveMigrationSubtaskRequest=GoogleCloudBigqueryMigrationV2ReceiveMigrationSubtaskResponseFrG   rS   rU   r,   r*   rV   rW   c   s6    H4J4Ji])*&'EThZ5r,   c                 D    U R                  S5      nU R                  X1US9$ )aF  Renew a lease for a migration subtask by subtask type.

Args:
  request: (BigquerymigrationProjectsLocationsSubtaskTypesRenewSubtaskLeaseRequest) input message
  global_params: (StandardQueryParameters, default: None) global arguments
Returns:
  (GoogleCloudBigqueryMigrationV2RenewSubtaskLeaseResponse) The response message.
RenewSubtaskLeaser8   r:   r=   s       r*   ra   JBigquerymigrationV2.ProjectsLocationsSubtaskTypesService.RenewSubtaskLeaseq   s0     ##$78f__
  8 8r,   c                  B    [         R                  " SSSS/S// SSSSS	S
9$ )Nz`v2/projects/{projectsId}/locations/{locationsId}/subtaskTypes/{subtaskTypesId}:renewSubtaskLeaserB   zCbigquerymigration.projects.locations.subtaskTypes.renewSubtaskLeaserC   z#v2/{+subtaskType}:renewSubtaskLease6googleCloudBigqueryMigrationV2RenewSubtaskLeaseRequestFBigquerymigrationProjectsLocationsSubtaskTypesRenewSubtaskLeaseRequest7GoogleCloudBigqueryMigrationV2RenewSubtaskLeaseResponseFrG   rS   rU   r,   r*   rV   rW   ~   s4    h.D.DtW%"O;NbT/r,   r2   r1   )__name__
__module____qualname____firstlineno____doc___NAMEr   r7   method_configrY   ra   __static_attributes____classcell__r)   s   @r*   r   r.   1   sB    I-E
8#M8-)8'##r,   r   c                   ~   ^  \ rS rSrSrSrU 4S jrSS jrS \l        SS jr	S \	l        SS	 jr
S
 \
l        SrU =r$ )5BigquerymigrationV2.ProjectsLocationsTaskTypesService   z<Service class for the projects_locations_taskTypes resource.r   c                 N   > [         [        R                  U ]  U5        0 U l        g r1   )r   r   r   r   r2   r3   s     r*   r   >BigquerymigrationV2.ProjectsLocationsTaskTypesService.__init__   $    AA4QRXYdr,   c                 D    U R                  S5      nU R                  X1US9$ )a  Updates a migration task's state via the task type. Transition from ORCHESTRATING to either RUNNING or FAILED, depending on the provided state, i.e. whether orchrestration was successful or not.

Args:
  request: (BigquerymigrationProjectsLocationsTaskTypesFinishMigrationTaskOrchestrationRequest) input message
  global_params: (StandardQueryParameters, default: None) global arguments
Returns:
  (GoogleProtobufEmpty) The response message.
 FinishMigrationTaskOrchestrationr8   r:   r=   s       r*   ry   VBigquerymigrationV2.ProjectsLocationsTaskTypesService.FinishMigrationTaskOrchestration   s0     ##$FGf__
  8 8r,   c                  B    [         R                  " SSSS/S// SSSSS	S
9$ )Nziv2/projects/{projectsId}/locations/{locationsId}/taskTypes/{taskTypesId}:finishMigrationTaskOrchestrationrB   zObigquerymigration.projects.locations.taskTypes.finishMigrationTaskOrchestrationtaskTypez/v2/{+taskType}:finishMigrationTaskOrchestrationEgoogleCloudBigqueryMigrationV2FinishMigrationTaskOrchestrationRequestRBigquerymigrationProjectsLocationsTaskTypesFinishMigrationTaskOrchestrationRequestrF   FrG   rS   rU   r,   r*   rV   >BigquerymigrationV2.ProjectsLocationsTaskTypesService.<lambda>   s4    X=S=S}c"|LG]n0>r,   c                 D    U R                  S5      nU R                  X1US9$ )a?  Assigns a migration task to the caller (usually an Orchestrator), if available. Example: `/v2/projects/123/locations/us/taskTypes:receiveMigrationTask`. The caller must define at least one type in the request parameters. The service creates a lease and marks the newly assigned task as ORCHESTRATING.

Args:
  request: (BigquerymigrationProjectsLocationsTaskTypesReceiveMigrationTaskRequest) input message
  global_params: (StandardQueryParameters, default: None) global arguments
Returns:
  (GoogleCloudBigqueryMigrationV2ReceiveMigrationTaskResponse) The response message.
ReceiveMigrationTaskr8   r:   r=   s       r*   r   JBigquerymigrationV2.ProjectsLocationsTaskTypesService.ReceiveMigrationTask   s0     ##$:;f__
  8 8r,   c                  B    [         R                  " SSSS/S// SSSSS	S
9$ )NzOv2/projects/{projectsId}/locations/{locationsId}/taskTypes:receiveMigrationTaskrB   zCbigquerymigration.projects.locations.taskTypes.receiveMigrationTaskr\   z4v2/{+projectLocation}/taskTypes:receiveMigrationTask9googleCloudBigqueryMigrationV2ReceiveMigrationTaskRequestFBigquerymigrationProjectsLocationsTaskTypesReceiveMigrationTaskRequest:GoogleCloudBigqueryMigrationV2ReceiveMigrationTaskResponseFrG   rS   rU   r,   r*   rV   r      s6    1G1GcW)*&'LQbW2r,   c                 D    U R                  S5      nU R                  X1US9$ )a7  Renew a lease for a migration task by task type.

Args:
  request: (BigquerymigrationProjectsLocationsTaskTypesRenewTaskLeaseRequest) input message
  global_params: (StandardQueryParameters, default: None) global arguments
Returns:
  (GoogleCloudBigqueryMigrationV2RenewTaskLeaseResponse) The response message.
RenewTaskLeaser8   r:   r=   s       r*   r   DBigquerymigrationV2.ProjectsLocationsTaskTypesService.RenewTaskLease   0     ##$45f__
  8 8r,   c                  B    [         R                  " SSSS/S// SSSSS	S
9$ )NzWv2/projects/{projectsId}/locations/{locationsId}/taskTypes/{taskTypesId}:renewTaskLeaserB   z=bigquerymigration.projects.locations.taskTypes.renewTaskLeaser|   zv2/{+taskType}:renewTaskLease3googleCloudBigqueryMigrationV2RenewTaskLeaseRequest@BigquerymigrationProjectsLocationsTaskTypesRenewTaskLeaseRequest4GoogleCloudBigqueryMigrationV2RenewTaskLeaseResponseFrG   rS   rU   r,   r*   rV   r      s4    8+A+AkQ"|L5K\Q,r,   rg   r1   )rh   ri   rj   rk   rl   rm   r   ry   rn   r   r   ro   rp   rq   s   @r*   r   rs      sB    F*E
86$28*&8$N  r,   r   c                   d   ^  \ rS rSrSrSrU 4S jrS
S jrS \l        S
S jr	S \	l        S	r
U =r$ )=BigquerymigrationV2.ProjectsLocationsWorkflowsSubtasksService   zEService class for the projects_locations_workflows_subtasks resource.r    c                 N   > [         [        R                  U ]  U5        0 U l        g r1   )r   r   r   r   r2   r3   s     r*   r   FBigquerymigrationV2.ProjectsLocationsWorkflowsSubtasksService.__init__   s$    II4YZ`adr,   c                 D    U R                  S5      nU R                  X1US9$ )a*  Gets a previously created migration subtask.

Args:
  request: (BigquerymigrationProjectsLocationsWorkflowsSubtasksGetRequest) input message
  global_params: (StandardQueryParameters, default: None) global arguments
Returns:
  (GoogleCloudBigqueryMigrationV2MigrationSubtask) The response message.
Getr8   r:   r=   s       r*   r   ABigquerymigrationV2.ProjectsLocationsWorkflowsSubtasksService.Get   /     ##E*f__
  8 8r,   c                  D    [         R                  " SSSS/S/S/SSSS	S
S9$ )Nz^v2/projects/{projectsId}/locations/{locationsId}/workflows/{workflowsId}/subtasks/{subtasksId}GETz;bigquerymigration.projects.locations.workflows.subtasks.getnamereadMask
v2/{+name} =BigquerymigrationProjectsLocationsWorkflowsSubtasksGetRequest.GoogleCloudBigqueryMigrationV2MigrationSubtaskFrG   rS   rU   r,   r*   rV   FBigquerymigrationV2.ProjectsLocationsWorkflowsSubtasksService.<lambda>   s6     6 6rOxH \"YK!r,   c                 D    U R                  S5      nU R                  X1US9$ )a8  Lists previously created migration subtasks.

Args:
  request: (BigquerymigrationProjectsLocationsWorkflowsSubtasksListRequest) input message
  global_params: (StandardQueryParameters, default: None) global arguments
Returns:
  (GoogleCloudBigqueryMigrationV2ListMigrationSubtasksResponse) The response message.
Listr8   r:   r=   s       r*   r   BBigquerymigrationV2.ProjectsLocationsWorkflowsSubtasksService.List  /     ##F+f__
  8 8r,   c                  F    [         R                  " SSSS/S// SQSSSS	S
S9$ )NzQv2/projects/{projectsId}/locations/{locationsId}/workflows/{workflowsId}/subtasksr   z<bigquerymigration.projects.locations.workflows.subtasks.listparent)filterpageSize	pageTokenr   zv2/{+parent}/subtasksr   >BigquerymigrationProjectsLocationsWorkflowsSubtasksListRequest;GoogleCloudBigqueryMigrationV2ListMigrationSubtasksResponseFrG   rS   rU   r,   r*   rV   r     s4    !7!7eP zJD-ZX"r,   rg   r1   )rh   ri   rj   rk   rl   rm   r   r   rn   r   ro   rp   rq   s   @r*   r   r      s1    O3E
8C8Dr,   r   c                      ^  \ rS rSrSrSrU 4S jrSS jrS \l        SS jr	S \	l        SS	 jr
S
 \
l        SS jrS \l        SS jrS \l        SrU =r$ )5BigquerymigrationV2.ProjectsLocationsWorkflowsServicei'  z<Service class for the projects_locations_workflows resource.r"   c                 N   > [         [        R                  U ]  U5        0 U l        g r1   )r   r   r!   r   r2   r3   s     r*   r   >BigquerymigrationV2.ProjectsLocationsWorkflowsService.__init__,  rw   r,   c                 D    U R                  S5      nU R                  X1US9$ )a  Creates a migration workflow.

Args:
  request: (BigquerymigrationProjectsLocationsWorkflowsCreateRequest) input message
  global_params: (StandardQueryParameters, default: None) global arguments
Returns:
  (GoogleCloudBigqueryMigrationV2MigrationWorkflow) The response message.
Creater8   r:   r=   s       r*   r   <BigquerymigrationV2.ProjectsLocationsWorkflowsService.Create1  /     ##H-f__
  8 8r,   c                  B    [         R                  " SSSS/S// SSSSS	S
9$ )N:v2/projects/{projectsId}/locations/{locationsId}/workflowsrB   z5bigquerymigration.projects.locations.workflows.creater   v2/{+parent}/workflows/googleCloudBigqueryMigrationV2MigrationWorkflow8BigquerymigrationProjectsLocationsWorkflowsCreateRequest/GoogleCloudBigqueryMigrationV2MigrationWorkflowFrG   rS   rU   r,   r*   rV   >BigquerymigrationV2.ProjectsLocationsWorkflowsService.<lambda>>  s4    8#9#9NI zJ.GTL$r,   c                 D    U R                  S5      nU R                  X1US9$ )a  Deletes a migration workflow by name.

Args:
  request: (BigquerymigrationProjectsLocationsWorkflowsDeleteRequest) input message
  global_params: (StandardQueryParameters, default: None) global arguments
Returns:
  (GoogleProtobufEmpty) The response message.
Deleter8   r:   r=   s       r*   r   <BigquerymigrationV2.ProjectsLocationsWorkflowsService.DeleteL  r   r,   c                  B    [         R                  " SSSS/S// SSSSS	S
9$ )NHv2/projects/{projectsId}/locations/{locationsId}/workflows/{workflowsId}DELETEz5bigquerymigration.projects.locations.workflows.deleter   r   r   8BigquerymigrationProjectsLocationsWorkflowsDeleteRequestrF   FrG   rS   rU   r,   r*   rV   r   Y  s4    8#9#9\IxH"T0$r,   c                 D    U R                  S5      nU R                  X1US9$ )a$  Gets a previously created migration workflow.

Args:
  request: (BigquerymigrationProjectsLocationsWorkflowsGetRequest) input message
  global_params: (StandardQueryParameters, default: None) global arguments
Returns:
  (GoogleCloudBigqueryMigrationV2MigrationWorkflow) The response message.
r   r8   r:   r=   s       r*   r   9BigquerymigrationV2.ProjectsLocationsWorkflowsService.Getg  r   r,   c                  D    [         R                  " SSSS/S/S/SSSS	S
S9$ )Nr   r   z2bigquerymigration.projects.locations.workflows.getr   r   r   r   5BigquerymigrationProjectsLocationsWorkflowsGetRequestr   FrG   rS   rU   r,   r*   rV   r   t  s6     6 6\FxH \"QL!r,   c                 D    U R                  S5      nU R                  X1US9$ )a1  Lists previously created migration workflow.

Args:
  request: (BigquerymigrationProjectsLocationsWorkflowsListRequest) input message
  global_params: (StandardQueryParameters, default: None) global arguments
Returns:
  (GoogleCloudBigqueryMigrationV2ListMigrationWorkflowsResponse) The response message.
r   r8   r:   r=   s       r*   r   :BigquerymigrationV2.ProjectsLocationsWorkflowsService.List  r   r,   c                  F    [         R                  " SSSS/S// SQSSSS	S
S9$ )Nr   r   z3bigquerymigration.projects.locations.workflows.listr   )r   r   r   r   r   6BigquerymigrationProjectsLocationsWorkflowsListRequest<GoogleCloudBigqueryMigrationV2ListMigrationWorkflowsResponseFrG   rS   rU   r,   r*   rV   r     s4    !7!7NG zJ:.RY"r,   c                 D    U R                  S5      nU R                  X1US9$ )a  Starts a previously created migration workflow. I.e., the state transitions from DRAFT to RUNNING. This is a no-op if the state is already RUNNING. An error will be signaled if the state is anything other than DRAFT or RUNNING.

Args:
  request: (BigquerymigrationProjectsLocationsWorkflowsStartRequest) input message
  global_params: (StandardQueryParameters, default: None) global arguments
Returns:
  (GoogleProtobufEmpty) The response message.
Startr8   r:   r=   s       r*   r   ;BigquerymigrationV2.ProjectsLocationsWorkflowsService.Start  s/     ##G,f__
  8 8r,   c                  B    [         R                  " SSSS/S// SSSSS	S
9$ )NzNv2/projects/{projectsId}/locations/{locationsId}/workflows/{workflowsId}:startrB   z4bigquerymigration.projects.locations.workflows.startr   zv2/{+name}:start;googleCloudBigqueryMigrationV2StartMigrationWorkflowRequest7BigquerymigrationProjectsLocationsWorkflowsStartRequestrF   FrG   rS   rU   r,   r*   rV   r     s4    ("8"8bHxH(SS0#r,   rg   r1   )rh   ri   rj   rk   rl   rm   r   r   rn   r   r   r   r   ro   rp   rq   s   @r*   r!   r   '  s^    F*E
8F8F8C8D8Er,   r!   c                   J   ^  \ rS rSrSrSrU 4S jrSS jrS \l        Sr	U =r
$ )	,BigquerymigrationV2.ProjectsLocationsServicei  z2Service class for the projects_locations resource.r$   c                 N   > [         [        R                  U ]  U5        0 U l        g r1   )r   r   r#   r   r2   r3   s     r*   r   5BigquerymigrationV2.ProjectsLocationsService.__init__  s#    88$HPdr,   c                 D    U R                  S5      nU R                  X1US9$ )a9  Translates input queries from source dialects to GoogleSQL.

Args:
  request: (BigquerymigrationProjectsLocationsTranslateQueryRequest) input message
  global_params: (StandardQueryParameters, default: None) global arguments
Returns:
  (GoogleCloudBigqueryMigrationV2TranslateQueryResponse) The response message.
TranslateQueryr8   r:   r=   s       r*   r   ;BigquerymigrationV2.ProjectsLocationsService.TranslateQuery  r   r,   c                  B    [         R                  " SSSS/S// SSSSS	S
9$ )Nz?v2/projects/{projectsId}/locations/{locationsId}:translateQueryrB   z3bigquerymigration.projects.locations.translateQueryr   zv2/{+parent}:translateQuery3googleCloudBigqueryMigrationV2TranslateQueryRequest7BigquerymigrationProjectsLocationsTranslateQueryRequest4GoogleCloudBigqueryMigrationV2TranslateQueryResponseFrG   rS   rU   r,   r*   rV   5BigquerymigrationV2.ProjectsLocationsService.<lambda>  s4    8+A+ASG zJ3KSQ,r,   rg   r1   )rh   ri   rj   rk   rl   rm   r   r   rn   ro   rp   rq   s   @r*   r#   r     s"    < E
8$N  r,   r#   c                   0   ^  \ rS rSrSrSrU 4S jrSrU =r$ )#BigquerymigrationV2.ProjectsServicei  z(Service class for the projects resource.r&   c                 N   > [         [        R                  U ]  U5        0 U l        g r1   )r   r   r%   r   r2   r3   s     r*   r   ,BigquerymigrationV2.ProjectsService.__init__  s#    //?Gdr,   rg   )	rh   ri   rj   rk   rl   rm   r   ro   rp   rq   s   @r*   r%   r     s    2E r,   r%   )r&   r$   r   r   r"   r    )r   NTNNFFNNNN)rh   ri   rj   rk   rl   messagesMESSAGES_MODULEr   MTLS_BASE_URL_PACKAGE_SCOPES_VERSION
_CLIENT_ID_CLIENT_SECRET_USER_AGENT_CLIENT_CLASS_NAME_URL_VERSION_API_KEYr   r   BaseApiServicer   r   r   r!   r#   r%   ro   rp   rq   s   @r*   r   r   
   s    J/8(B- (=>'(*"."+,,()-6:/4<@?C	/,YX-D-D YvY(*A*A Yv>(2I2I >@O(*A*A Ob#!8!8 #J//  r,   r   N)
rl   
__future__r   apitools.base.pyr   :googlecloudsdk.generated_clients.apis.bigquerymigration.v2r   r   BaseApiClientr   rU   r,   r*   <module>r      s'    @ ' % p[(00 [r,   