
    7                     V    d Z ddlmZ ddlmZ ddlmZ  G d dej                        Z	y)z7Generated client library for composer version v1alpha2.    )absolute_import)base_api)composer_v1alpha2_messagesc                   X    e Zd ZdZeZdZdZdZdgZ	dZ
dZdZd	Zd ZdZd
Z	 	 	 	 	 d$ fd	Z G d dej&                        Z G d dej&                        Z G d dej&                        Z G d dej&                        Z G d dej&                        Z G d dej&                        Z G d dej&                        Z G d dej&                        Z G d dej&                        Z G d dej&                        Z G d  d!ej&                        Z G d" d#ej&                        Z xZ S )%ComposerV1alpha2z?Generated client library for service composer version v1alpha2.z https://composer.googleapis.com/z%https://composer.mtls.googleapis.com/composerz.https://www.googleapis.com/auth/cloud-platformv1alpha2	CLIENT_IDCLIENT_SECRETzgoogle-cloud-sdkNc                 r   |xs | j                   }t        t        |   |||||||||	|
|       | j	                  |       | _        | j                  |       | _        | j                  |       | _	        | j                  |       | _        | j                  |       | _        | j                  |       | _        | j!                  |       | _        | j%                  |       | _        | j)                  |       | _        | j-                  |       | _        | j1                  |       | _        | j5                  |       | _        y)zCreate a new composer handle.)
credentialsget_credentialshttpmodellog_requestlog_responsecredentials_argsdefault_global_paramsadditional_http_headersresponse_encodingN)BASE_URLsuperr   __init__<ProjectsLocationsEnvironmentsDagsDagRunsTaskInstancesService:projects_locations_environments_dags_dagRuns_taskInstances/ProjectsLocationsEnvironmentsDagsDagRunsService,projects_locations_environments_dags_dagRuns-ProjectsLocationsEnvironmentsDagsTasksService*projects_locations_environments_dags_tasks(ProjectsLocationsEnvironmentsDagsService$projects_locations_environments_dags;ProjectsLocationsEnvironmentsUserWorkloadsConfigMapsService7projects_locations_environments_userWorkloadsConfigMaps8ProjectsLocationsEnvironmentsUserWorkloadsSecretsService4projects_locations_environments_userWorkloadsSecrets-ProjectsLocationsEnvironmentsWorkloadsService)projects_locations_environments_workloads$ProjectsLocationsEnvironmentsServiceprojects_locations_environments%ProjectsLocationsImageVersionsService projects_locations_imageVersions"ProjectsLocationsOperationsServiceprojects_locations_operationsProjectsLocationsServiceprojects_locationsProjectsServiceprojects)selfurlr   r   r   r   r   r   r   r   r   r   	__class__s               Wlib/googlecloudsdk/generated_clients/apis/composer/v1alpha2/composer_v1alpha2_client.pyr   zComposerV1alpha2.__init__   s^    
C	
D*'d%l)3 7+ + - GK  GH  GH  IM  GNDC8<8l8lmq8rD56:6h6him6nD3040]0]^b0cD-CG  DD  DD  EI  DJD@@D@}@}  C  ADD=595g5ghl5mD2+/+T+TUY+ZD(,0,V,VW[,\D))-)P)PQU)VD&";;DAD((.DM    c                   V     e Zd ZdZdZ fdZddZd e_        ddZd e_         xZ	S )	MComposerV1alpha2.ProjectsLocationsEnvironmentsDagsDagRunsTaskInstancesServicezZService class for the projects_locations_environments_dags_dagRuns_taskInstances resource.r   c                 N    t         t        j                  |   |       i | _        y N)r   r   r   r   _upload_configsr2   clientr4   s     r5   r   zVComposerV1alpha2.ProjectsLocationsEnvironmentsDagsDagRunsTaskInstancesService.__init__<   s%    YY[_ijpqdr6   c                 L    | j                  d      }| j                  |||      S )a$  Retrieves a task instance.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsDagsDagRunsTaskInstancesGetRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (TaskInstance) The response message.
      Getglobal_paramsGetMethodConfig
_RunMethodr2   requestrA   configs       r5   r?   zQComposerV1alpha2.ProjectsLocationsEnvironmentsDagsDagRunsTaskInstancesService.GetA   1     ##E*f__
'  8 8r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nzv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}/dags/{dagsId}/dagRuns/{dagRunsId}/taskInstances/{taskInstancesId}GETzGcomposer.projects.locations.environments.dags.dagRuns.taskInstances.getnamev1alpha2/{+name} GComposerProjectsLocationsEnvironmentsDagsDagRunsTaskInstancesGetRequestTaskInstanceF	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 r6   r5   <lambda>zVComposerV1alpha2.ProjectsLocationsEnvironmentsDagsDagRunsTaskInstancesService.<lambda>N   s7     6 6 k[xH(c)!r6   c                 L    | j                  d      }| j                  |||      S )aE  Lists task instances for a specified DAG run.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsDagsDagRunsTaskInstancesListRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (ListTaskInstancesResponse) The response message.
      Listr@   rB   rE   s       r5   ra   zRComposerV1alpha2.ProjectsLocationsEnvironmentsDagsDagRunsTaskInstancesService.List\   1     ##F+f__
'  8 8r6   c                  J    t        j                  ddddgdgg ddddd	d
      S )Nzv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}/dags/{dagsId}/dagRuns/{dagRunsId}/taskInstancesrJ   zHcomposer.projects.locations.environments.dags.dagRuns.taskInstances.listparentfilterpageSize	pageTokenz v1alpha2/{+parent}/taskInstancesrM   HComposerProjectsLocationsEnvironmentsDagsDagRunsTaskInstancesListRequestListTaskInstancesResponseFrP   r\   r^   r6   r5   r_   zVComposerV1alpha2.ProjectsLocationsEnvironmentsDagsDagRunsTaskInstancesService.<lambda>i   s7    !7!7 Y\ zJ88d6"r6   r:   
__name__
__module____qualname____doc___NAMEr   r?   method_configra   __classcell__r4   s   @r5   r   r8   7   s1    dHE
8C8Dr6   r   c                   V     e Zd ZdZdZ fdZddZd e_        ddZd e_         xZ	S )	@ComposerV1alpha2.ProjectsLocationsEnvironmentsDagsDagRunsServicezLService class for the projects_locations_environments_dags_dagRuns resource.r   c                 N    t         t        j                  |   |       i | _        y r:   )r   r   r   r   r;   r<   s     r5   r   zIComposerV1alpha2.ProjectsLocationsEnvironmentsDagsDagRunsService.__init__|   s$    LLd\]cddr6   c                 L    | j                  d      }| j                  |||      S )a  Retrieves a DAG run.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsDagsDagRunsGetRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (DagRun) The response message.
      r?   r@   rB   rE   s       r5   r?   zDComposerV1alpha2.ProjectsLocationsEnvironmentsDagsDagRunsService.Get   rH   r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nzvv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}/dags/{dagsId}/dagRuns/{dagRunsId}rJ   z9composer.projects.locations.environments.dags.dagRuns.getrK   rL   rM   :ComposerProjectsLocationsEnvironmentsDagsDagRunsGetRequestDagRunFrP   r\   r^   r6   r5   r_   zIComposerV1alpha2.ProjectsLocationsEnvironmentsDagsDagRunsService.<lambda>   s7     6 6 KMxH(V#!r6   c                 L    | j                  d      }| j                  |||      S )a  Lists DAG runs of a DAG.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsDagsDagRunsListRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (ListDagRunsResponse) The response message.
      ra   r@   rB   rE   s       r5   ra   zEComposerV1alpha2.ProjectsLocationsEnvironmentsDagsDagRunsService.List   rb   r6   c                  J    t        j                  ddddgdgg ddddd	d
      S )Nzjv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}/dags/{dagsId}/dagRunsrJ   z:composer.projects.locations.environments.dags.dagRuns.listrd   )rf   rg   rh   runsCountPerDagzv1alpha2/{+parent}/dagRunsrM   ;ComposerProjectsLocationsEnvironmentsDagsDagRunsListRequestListDagRunsResponseFrP   r\   r^   r6   r5   r_   zIComposerV1alpha2.ProjectsLocationsEnvironmentsDagsDagRunsService.<lambda>   s4    !7!7~N zJK2W0"r6   r:   rk   rs   s   @r5   r   ru   w   s1    V:E
8C8Dr6   r   c                   >     e Zd ZdZdZ fdZddZd e_         xZS )>ComposerV1alpha2.ProjectsLocationsEnvironmentsDagsTasksServicezJService class for the projects_locations_environments_dags_tasks resource.r   c                 N    t         t        j                  |   |       i | _        y r:   )r   r   r   r   r;   r<   s     r5   r   zGComposerV1alpha2.ProjectsLocationsEnvironmentsDagsTasksService.__init__   $    JJDZ[abdr6   c                 L    | j                  d      }| j                  |||      S )a  Lists tasks of a DAG.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsDagsTasksListRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (ListTasksResponse) The response message.
      ra   r@   rB   rE   s       r5   ra   zCComposerV1alpha2.ProjectsLocationsEnvironmentsDagsTasksService.List   rb   r6   c                  J    t        j                  ddddgdgddgddd	d
d      S )Nzhv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}/dags/{dagsId}/tasksrJ   z8composer.projects.locations.environments.dags.tasks.listrd   rg   rh   zv1alpha2/{+parent}/tasksrM   9ComposerProjectsLocationsEnvironmentsDagsTasksListRequestListTasksResponseFrP   r\   r^   r6   r5   r_   zGComposerV1alpha2.ProjectsLocationsEnvironmentsDagsTasksService.<lambda>   s9    !7!7|L zJ +.0U."r6   r:   	rl   rm   rn   ro   rp   r   ra   rq   rr   rs   s   @r5   r   r      s"    T8E
8Dr6   r   c                        e Zd ZdZdZ fdZddZd e_        ddZd e_        ddZ	d	 e	_        dd
Z
d e
_        ddZd e_        ddZd e_        ddZd e_         xZS )9ComposerV1alpha2.ProjectsLocationsEnvironmentsDagsServicezDService class for the projects_locations_environments_dags resource.r!   c                 N    t         t        j                  |   |       i | _        y r:   )r   r   r    r   r;   r<   s     r5   r   zBComposerV1alpha2.ProjectsLocationsEnvironmentsDagsService.__init__   s$    EEtUV\]dr6   c                 L    | j                  d      }| j                  |||      S )a  Activates a dag.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsDagsActivateRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (Dag) The response message.
      Activater@   rB   rE   s       r5   r   zBComposerV1alpha2.ProjectsLocationsEnvironmentsDagsService.Activate   s1     ##J/f__
'  8 8r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nzkv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}/dags/{dagsId}:activatePOSTz6composer.projects.locations.environments.dags.activaterK   zv1alpha2/{+name}:activateactivateDagRequest8ComposerProjectsLocationsEnvironmentsDagsActivateRequestDagFrP   r\   r^   r6   r5   r_   zBComposerV1alpha2.ProjectsLocationsEnvironmentsDagsService.<lambda>   s4    X%;%;JxH1*T &r6   c                 L    | j                  d      }| j                  |||      S )zRetrieves a DAG.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsDagsGetRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (Dag) The response message.
      r?   r@   rB   rE   s       r5   r?   z=ComposerV1alpha2.ProjectsLocationsEnvironmentsDagsService.Get  rH   r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nzbv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}/dags/{dagsId}rJ   z1composer.projects.locations.environments.dags.getrK   rL   rM   3ComposerProjectsLocationsEnvironmentsDagsGetRequestr   FrP   r\   r^   r6   r5   r_   zBComposerV1alpha2.ProjectsLocationsEnvironmentsDagsService.<lambda>  s4     6 6vExH(O !r6   c                 L    | j                  d      }| j                  |||      S )a  Retrieves DAG source code.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsDagsGetSourceCodeRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (SourceCode) The response message.
      GetSourceCoder@   rB   rE   s       r5   r   zGComposerV1alpha2.ProjectsLocationsEnvironmentsDagsService.GetSourceCode  s1     ##O4f__
'  8 8r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nzpv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}/dags/{dagsId}:getSourceCoderJ   z;composer.projects.locations.environments.dags.getSourceCodedagzv1alpha2/{+dag}:getSourceCoderM   =ComposerProjectsLocationsEnvironmentsDagsGetSourceCodeRequest
SourceCodeFrP   r\   r^   r6   r5   r_   zBComposerV1alpha2.ProjectsLocationsEnvironmentsDagsService.<lambda>)  s7    (*@*@ EOwG5Y'+r6   c                 L    | j                  d      }| j                  |||      S )a  Lists DAGs in an environment.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsDagsListRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (ListDagsResponse) The response message.
      ra   r@   rB   rE   s       r5   ra   z>ComposerV1alpha2.ProjectsLocationsEnvironmentsDagsService.List7  rb   r6   c                  J    t        j                  ddddgdgddgddd	d
d      S )NzYv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}/dagsrJ   z2composer.projects.locations.environments.dags.listrd   rg   rh   zv1alpha2/{+parent}/dagsrM   4ComposerProjectsLocationsEnvironmentsDagsListRequestListDagsResponseFrP   r\   r^   r6   r5   r_   zBComposerV1alpha2.ProjectsLocationsEnvironmentsDagsService.<lambda>D  s9    !7!7mF zJ +./P-"r6   c                 L    | j                  d      }| j                  |||      S )a8  List DAGs with statistics for a given time interval.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsDagsListStatsRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (ListDagStatsResponse) The response message.
      	ListStatsr@   rB   rE   s       r5   r   zCComposerV1alpha2.ProjectsLocationsEnvironmentsDagsService.ListStatsR  s1     ##K0f__
'  8 8r6   c                  J    t        j                  ddddgdgg ddddd	d
      S )Nzcv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}/dags:listStatsrJ   z7composer.projects.locations.environments.dags.listStatsenvironment)interval_endTimeinterval_startTimerg   rh   z&v1alpha2/{+environment}/dags:listStatsrM   9ComposerProjectsLocationsEnvironmentsDagsListStatsRequestListDagStatsResponseFrP   r\   r^   r6   r5   r_   zBComposerV1alpha2.ProjectsLocationsEnvironmentsDagsService.<lambda>_  s4    h&<&<wK%"OX>U1'r6   c                 L    | j                  d      }| j                  |||      S )zPauses a dag.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsDagsPauseRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (Dag) The response message.
      Pauser@   rB   rE   s       r5   r   z?ComposerV1alpha2.ProjectsLocationsEnvironmentsDagsService.Pausem  1     ##G,f__
'  8 8r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nzhv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}/dags/{dagsId}:pauser   z3composer.projects.locations.environments.dags.pauserK   zv1alpha2/{+name}:pausepauseDagRequest5ComposerProjectsLocationsEnvironmentsDagsPauseRequestr   FrP   r\   r^   r6   r5   r_   zBComposerV1alpha2.ProjectsLocationsEnvironmentsDagsService.<lambda>z  s4    ("8"8|GxH.'Q #r6   c                 L    | j                  d      }| j                  |||      S )a  Trigger a DAG run.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsDagsTriggerRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (DagRun) The response message.
      Triggerr@   rB   rE   s       r5   r   zAComposerV1alpha2.ProjectsLocationsEnvironmentsDagsService.Trigger  1     ##I.f__
'  8 8r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nzjv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}/dags/{dagsId}:triggerr   z5composer.projects.locations.environments.dags.triggerr   zv1alpha2/{+dag}:triggertriggerDagRequest7ComposerProjectsLocationsEnvironmentsDagsTriggerRequestrz   FrP   r\   r^   r6   r5   r_   zBComposerV1alpha2.ProjectsLocationsEnvironmentsDagsService.<lambda>  s4    H$:$:~IwG/)S#%r6   r:   )rl   rm   rn   ro   rp   r   r   rq   r?   r   ra   r   r   r   rr   rs   s   @r5   r    r      s|    N2E
8H8C8#M8D8I8E8Gr6   r    c                        e Zd ZdZdZ fdZddZd e_        ddZd e_        ddZ	d	 e	_        dd
Z
d e
_        ddZd e_         xZS )LComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsConfigMapsServicezWService class for the projects_locations_environments_userWorkloadsConfigMaps resource.r#   c                 N    t         t        j                  |   |       i | _        y r:   )r   r   r"   r   r;   r<   s     r5   r   zUComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsConfigMapsService.__init__  s%    XXZ^hiopdr6   c                 L    | j                  d      }| j                  |||      S )a  Creates a user workloads ConfigMap. This method is supported for Cloud Composer environments in versions composer-3-airflow-*.*.*-build.* and newer.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsUserWorkloadsConfigMapsCreateRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (UserWorkloadsConfigMap) The response message.
      Creater@   rB   rE   s       r5   r   zSComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsConfigMapsService.Create  1     ##H-f__
'  8 8r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nlv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}/userWorkloadsConfigMapsr   zGcomposer.projects.locations.environments.userWorkloadsConfigMaps.createrd   *v1alpha2/{+parent}/userWorkloadsConfigMapsuserWorkloadsConfigMapIComposerProjectsLocationsEnvironmentsUserWorkloadsConfigMapsCreateRequestUserWorkloadsConfigMapFrP   r\   r^   r6   r5   r_   zUComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsConfigMapsService.<lambda>  s7    8#9#9 A[ zJB.e3$r6   c                 L    | j                  d      }| j                  |||      S )a  Deletes a user workloads ConfigMap. This method is supported for Cloud Composer environments in versions composer-3-airflow-*.*.*-build.* and newer.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsUserWorkloadsConfigMapsDeleteRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (Empty) The response message.
      Deleter@   rB   rE   s       r5   r   zSComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsConfigMapsService.Delete  r   r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}/userWorkloadsConfigMaps/{userWorkloadsConfigMapsId}DELETEzGcomposer.projects.locations.environments.userWorkloadsConfigMaps.deleterK   rL   rM   IComposerProjectsLocationsEnvironmentsUserWorkloadsConfigMapsDeleteRequestEmptyFrP   r\   r^   r6   r5   r_   zUComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsConfigMapsService.<lambda>  s7    8#9#9 ][xH(e"$r6   c                 L    | j                  d      }| j                  |||      S )a  Gets an existing user workloads ConfigMap. This method is supported for Cloud Composer environments in versions composer-3-airflow-*.*.*-build.* and newer.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsUserWorkloadsConfigMapsGetRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (UserWorkloadsConfigMap) The response message.
      r?   r@   rB   rE   s       r5   r?   zPComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsConfigMapsService.Get  rH   r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nr   rJ   zDcomposer.projects.locations.environments.userWorkloadsConfigMaps.getrK   rL   rM   FComposerProjectsLocationsEnvironmentsUserWorkloadsConfigMapsGetRequestr   FrP   r\   r^   r6   r5   r_   zUComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsConfigMapsService.<lambda>  s7     6 6 ]XxH(b3!r6   c                 L    | j                  d      }| j                  |||      S )a  Lists user workloads ConfigMaps. This method is supported for Cloud Composer environments in versions composer-3-airflow-*.*.*-build.* and newer.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsUserWorkloadsConfigMapsListRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (ListUserWorkloadsConfigMapsResponse) The response message.
      ra   r@   rB   rE   s       r5   ra   zQComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsConfigMapsService.List  rb   r6   c                  J    t        j                  ddddgdgddgddd	d
d      S )Nr   rJ   zEcomposer.projects.locations.environments.userWorkloadsConfigMaps.listrd   rg   rh   r   rM   GComposerProjectsLocationsEnvironmentsUserWorkloadsConfigMapsListRequest#ListUserWorkloadsConfigMapsResponseFrP   r\   r^   r6   r5   r_   zUComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsConfigMapsService.<lambda>  s<    !7!7 AY zJ +.Bc@"r6   c                 L    | j                  d      }| j                  |||      S )aw  Updates a user workloads ConfigMap. This method is supported for Cloud Composer environments in versions composer-3-airflow-*.*.*-build.* and newer.

      Args:
        request: (UserWorkloadsConfigMap) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (UserWorkloadsConfigMap) The response message.
      Updater@   rB   rE   s       r5   r   zSComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsConfigMapsService.Update  r   r6   c                  F    t        j                  ddddgdgg ddddd	      S )
Nr   PUTzGcomposer.projects.locations.environments.userWorkloadsConfigMaps.updaterK   rL   	<request>r   FrP   r\   r^   r6   r5   r_   zUComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsConfigMapsService.<lambda>&  s7    8#9#9 ][xH(!23$r6   r:   rl   rm   rn   ro   rp   r   r   rq   r   r?   ra   r   rr   rs   s   @r5   r"   r     s^    aEE
8F8F8C8D8Fr6   r"   c                        e Zd ZdZdZ fdZddZd e_        ddZd e_        ddZ	d	 e	_        dd
Z
d e
_        ddZd e_         xZS )IComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsSecretsServicezTService class for the projects_locations_environments_userWorkloadsSecrets resource.r%   c                 N    t         t        j                  |   |       i | _        y r:   )r   r   r$   r   r;   r<   s     r5   r   zRComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsSecretsService.__init__9  s%    UUW[eflmdr6   c                 L    | j                  d      }| j                  |||      S )a  Creates a user workloads Secret. This method is supported for Cloud Composer environments in versions composer-3-airflow-*.*.*-build.* and newer.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsUserWorkloadsSecretsCreateRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (UserWorkloadsSecret) The response message.
      r   r@   rB   rE   s       r5   r   zPComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsSecretsService.Create>  r   r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Niv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}/userWorkloadsSecretsr   zDcomposer.projects.locations.environments.userWorkloadsSecrets.createrd   'v1alpha2/{+parent}/userWorkloadsSecretsuserWorkloadsSecretFComposerProjectsLocationsEnvironmentsUserWorkloadsSecretsCreateRequestUserWorkloadsSecretFrP   r\   r^   r6   r5   r_   zRComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsSecretsService.<lambda>K  s4    8#9#9}X zJ?+b0$r6   c                 L    | j                  d      }| j                  |||      S )a  Deletes a user workloads Secret. This method is supported for Cloud Composer environments in versions composer-3-airflow-*.*.*-build.* and newer.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsUserWorkloadsSecretsDeleteRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (Empty) The response message.
      r   r@   rB   rE   s       r5   r   zPComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsSecretsService.DeleteY  r   r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}/userWorkloadsSecrets/{userWorkloadsSecretsId}r   zDcomposer.projects.locations.environments.userWorkloadsSecrets.deleterK   rL   rM   FComposerProjectsLocationsEnvironmentsUserWorkloadsSecretsDeleteRequestr   FrP   r\   r^   r6   r5   r_   zRComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsSecretsService.<lambda>f  s7    8#9#9 WXxH(b"$r6   c                 L    | j                  d      }| j                  |||      S )a  Gets an existing user workloads Secret. Values of the "data" field in the response are cleared. This method is supported for Cloud Composer environments in versions composer-3-airflow-*.*.*-build.* and newer.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsUserWorkloadsSecretsGetRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (UserWorkloadsSecret) The response message.
      r?   r@   rB   rE   s       r5   r?   zMComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsSecretsService.Gett  rH   r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nr   rJ   zAcomposer.projects.locations.environments.userWorkloadsSecrets.getrK   rL   rM   CComposerProjectsLocationsEnvironmentsUserWorkloadsSecretsGetRequestr   FrP   r\   r^   r6   r5   r_   zRComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsSecretsService.<lambda>  s7     6 6 WUxH(_0!r6   c                 L    | j                  d      }| j                  |||      S )a  Lists user workloads Secrets. This method is supported for Cloud Composer environments in versions composer-3-airflow-*.*.*-build.* and newer.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsUserWorkloadsSecretsListRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (ListUserWorkloadsSecretsResponse) The response message.
      ra   r@   rB   rE   s       r5   ra   zNComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsSecretsService.List  rb   r6   c                  J    t        j                  ddddgdgddgddd	d
d      S )Nr   rJ   zBcomposer.projects.locations.environments.userWorkloadsSecrets.listrd   rg   rh   r   rM   DComposerProjectsLocationsEnvironmentsUserWorkloadsSecretsListRequest ListUserWorkloadsSecretsResponseFrP   r\   r^   r6   r5   r_   zRComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsSecretsService.<lambda>  s9    !7!7}V zJ +.?`="r6   c                 L    | j                  d      }| j                  |||      S )an  Updates a user workloads Secret. This method is supported for Cloud Composer environments in versions composer-3-airflow-*.*.*-build.* and newer.

      Args:
        request: (UserWorkloadsSecret) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (UserWorkloadsSecret) The response message.
      r   r@   rB   rE   s       r5   r   zPComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsSecretsService.Update  r   r6   c                  F    t        j                  ddddgdgg ddddd	      S )
Nr   r   zDcomposer.projects.locations.environments.userWorkloadsSecrets.updaterK   rL   r   r   FrP   r\   r^   r6   r5   r_   zRComposerV1alpha2.ProjectsLocationsEnvironmentsUserWorkloadsSecretsService.<lambda>  s7    8#9#9 WXxH(!/0$r6   r:   r   rs   s   @r5   r$   r   4  s^    ^BE
8F8F8C8D8Fr6   r$   c                   >     e Zd ZdZdZ fdZddZd e_         xZS )>ComposerV1alpha2.ProjectsLocationsEnvironmentsWorkloadsServicezIService class for the projects_locations_environments_workloads resource.r'   c                 N    t         t        j                  |   |       i | _        y r:   )r   r   r&   r   r;   r<   s     r5   r   zGComposerV1alpha2.ProjectsLocationsEnvironmentsWorkloadsService.__init__  r   r6   c                 L    | j                  d      }| j                  |||      S )a  Lists workloads in a Cloud Composer environment. Workload is a unit that runs a single Composer component. This method is supported for Cloud Composer environments in versions composer-2.*.*-airflow-*.*.* and newer.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsWorkloadsListRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (ListWorkloadsResponse) The response message.
      ra   r@   rB   rE   s       r5   ra   zCComposerV1alpha2.ProjectsLocationsEnvironmentsWorkloadsService.List  rb   r6   c                  J    t        j                  ddddgdgg ddddd	d
      S )Nz^v1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}/workloadsrJ   z7composer.projects.locations.environments.workloads.listrd   re   zv1alpha2/{+parent}/workloadsrM   9ComposerProjectsLocationsEnvironmentsWorkloadsListRequestListWorkloadsResponseFrP   r\   r^   r6   r5   r_   zGComposerV1alpha2.ProjectsLocationsEnvironmentsWorkloadsService.<lambda>  s4    !7!7rK zJ84U2"r6   r:   r   rs   s   @r5   r&   r     s"    S7E
8Dr6   r&   c                       e Zd ZdZdZ fdZd$dZd e_        d$dZd e_        d$dZ	d	 e	_        d$d
Z
d e
_        d$dZd e_        d$dZd e_        d$dZd e_        d$dZd e_        d$dZd e_        d$dZd e_        d$dZd e_        d$dZd e_        d$dZd e_        d$dZd e_        d$d Zd! e_        d$d"Zd# e_         xZS )%5ComposerV1alpha2.ProjectsLocationsEnvironmentsServicez?Service class for the projects_locations_environments resource.r)   c                 N    t         t        j                  |   |       i | _        y r:   )r   r   r(   r   r;   r<   s     r5   r   z>ComposerV1alpha2.ProjectsLocationsEnvironmentsService.__init__  s$    AA4QRXYdr6   c                 L    | j                  d      }| j                  |||      S )a  Check if an upgrade operation on the environment will succeed. In case of problems detailed info can be found in the returned Operation.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsCheckUpgradeRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (Operation) The response message.
      CheckUpgrader@   rB   rE   s       r5   r   zBComposerV1alpha2.ProjectsLocationsEnvironmentsService.CheckUpgrade  1     ##N3f__
'  8 8r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nzav1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}:checkUpgrader   z5composer.projects.locations.environments.checkUpgrader   z$v1alpha2/{+environment}:checkUpgradecheckUpgradeRequest8ComposerProjectsLocationsEnvironmentsCheckUpgradeRequest	OperationFrP   r\   r^   r6   r5   r_   z>ComposerV1alpha2.ProjectsLocationsEnvironmentsService.<lambda>  4    )?)?uI%"O<+T&*r6   c                 L    | j                  d      }| j                  |||      S )a  Create a new environment.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsCreateRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (Operation) The response message.
      r   r@   rB   rE   s       r5   r   z<ComposerV1alpha2.ProjectsLocationsEnvironmentsService.Create  r   r6   c                  F    t        j                  ddddgdgg ddddd	
      S )NCv1alpha2/projects/{projectsId}/locations/{locationsId}/environmentsr   z/composer.projects.locations.environments.createrd   v1alpha2/{+parent}/environmentsr   2ComposerProjectsLocationsEnvironmentsCreateRequestr   FrP   r\   r^   r6   r5   r_   z>ComposerV1alpha2.ProjectsLocationsEnvironmentsService.<lambda>  s4    8#9#9WC zJ7#N&$r6   c                 L    | j                  d      }| j                  |||      S )a@  Triggers database failover (only for highly resilient environments).

      Args:
        request: (ComposerProjectsLocationsEnvironmentsDatabaseFailoverRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (Operation) The response message.
      DatabaseFailoverr@   rB   rE   s       r5   r  zFComposerV1alpha2.ProjectsLocationsEnvironmentsService.DatabaseFailover*  2     ##$67f__
'  8 8r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nzev1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}:databaseFailoverr   z9composer.projects.locations.environments.databaseFailoverr   z(v1alpha2/{+environment}:databaseFailoverdatabaseFailoverRequest<ComposerProjectsLocationsEnvironmentsDatabaseFailoverRequestr   FrP   r\   r^   r6   r5   r_   z>ComposerV1alpha2.ProjectsLocationsEnvironmentsService.<lambda>7  s4    X-C-CyM%"O@/X&.r6   c                 L    | j                  d      }| j                  |||      S )a  Delete an environment.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsDeleteRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (Operation) The response message.
      r   r@   rB   rE   s       r5   r   z<ComposerV1alpha2.ProjectsLocationsEnvironmentsService.DeleteE  r   r6   c                  F    t        j                  ddddgdgg ddddd	
      S )NTv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}r   z/composer.projects.locations.environments.deleterK   rL   rM   2ComposerProjectsLocationsEnvironmentsDeleteRequestr   FrP   r\   r^   r6   r5   r_   z>ComposerV1alpha2.ProjectsLocationsEnvironmentsService.<lambda>R  s4    8#9#9hCxH(N&$r6   c                 L    | j                  d      }| j                  |||      S )a2  Executes Airflow CLI command.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsExecuteAirflowCommandRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (ExecuteAirflowCommandResponse) The response message.
      ExecuteAirflowCommandr@   rB   rE   s       r5   r  zKComposerV1alpha2.ProjectsLocationsEnvironmentsService.ExecuteAirflowCommand`  s2     ##$;<f__
'  8 8r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nzjv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}:executeAirflowCommandr   z>composer.projects.locations.environments.executeAirflowCommandr   z-v1alpha2/{+environment}:executeAirflowCommandexecuteAirflowCommandRequestAComposerProjectsLocationsEnvironmentsExecuteAirflowCommandRequestExecuteAirflowCommandResponseFrP   r\   r^   r6   r5   r_   z>ComposerV1alpha2.ProjectsLocationsEnvironmentsService.<lambda>m  s4    (2H2H~R%"OE4]:3r6   c                 L    | j                  d      }| j                  |||      S )a5  Fetches database properties.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsFetchDatabasePropertiesRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (FetchDatabasePropertiesResponse) The response message.
      FetchDatabasePropertiesr@   rB   rE   s       r5   r  zMComposerV1alpha2.ProjectsLocationsEnvironmentsService.FetchDatabaseProperties{  s2     ##$=>f__
'  8 8r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nzlv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}:fetchDatabasePropertiesrJ   z@composer.projects.locations.environments.fetchDatabasePropertiesr   z/v1alpha2/{+environment}:fetchDatabasePropertiesrM   CComposerProjectsLocationsEnvironmentsFetchDatabasePropertiesRequestFetchDatabasePropertiesResponseFrP   r\   r^   r6   r5   r_   z>ComposerV1alpha2.ProjectsLocationsEnvironmentsService.<lambda>  s7    H4J4J AT%"OG_<5r6   c                 L    | j                  d      }| j                  |||      S )a  Get an existing environment.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsGetRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (Environment) The response message.
      r?   r@   rB   rE   s       r5   r?   z9ComposerV1alpha2.ProjectsLocationsEnvironmentsService.Get  rH   r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nr
  rJ   z,composer.projects.locations.environments.getrK   rL   rM   /ComposerProjectsLocationsEnvironmentsGetRequestEnvironmentFrP   r\   r^   r6   r5   r_   z>ComposerV1alpha2.ProjectsLocationsEnvironmentsService.<lambda>  s4     6 6h@xH(K(!r6   c                 L    | j                  d      }| j                  |||      S )a  List environments.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsListRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (ListEnvironmentsResponse) The response message.
      ra   r@   rB   rE   s       r5   ra   z:ComposerV1alpha2.ProjectsLocationsEnvironmentsService.List  rb   r6   c                  J    t        j                  ddddgdgddgddd	d
d      S )Nr   rJ   z-composer.projects.locations.environments.listrd   rg   rh   r   rM   0ComposerProjectsLocationsEnvironmentsListRequestListEnvironmentsResponseFrP   r\   r^   r6   r5   r_   z>ComposerV1alpha2.ProjectsLocationsEnvironmentsService.<lambda>  s9    !7!7WA zJ +.7L5"r6   c                 L    | j                  d      }| j                  |||      S )a#  Lists DAG import errors.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsListImportErrorsRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (ListImportErrorsResponse) The response message.
      ListImportErrorsr@   rB   rE   s       r5   r   zFComposerV1alpha2.ProjectsLocationsEnvironmentsService.ListImportErrors  r  r6   c                  J    t        j                  ddddgdgddgddd	d
d      S )Nzev1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}:listImportErrorsrJ   z9composer.projects.locations.environments.listImportErrorsr   rg   rh   z(v1alpha2/{+environment}:listImportErrorsrM   <ComposerProjectsLocationsEnvironmentsListImportErrorsRequestListImportErrorsResponseFrP   r\   r^   r6   r5   r_   z>ComposerV1alpha2.ProjectsLocationsEnvironmentsService.<lambda>  s9    X-C-CyM%"O +.@X5.r6   c                 L    | j                  d      }| j                  |||      S )a  Loads a snapshot of a Cloud Composer environment. As a result of this operation, a snapshot of environment's specified in LoadSnapshotRequest is loaded into the environment.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsLoadSnapshotRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (Operation) The response message.
      LoadSnapshotr@   rB   rE   s       r5   r%  zBComposerV1alpha2.ProjectsLocationsEnvironmentsService.LoadSnapshot  r   r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nzav1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}:loadSnapshotr   z5composer.projects.locations.environments.loadSnapshotr   z$v1alpha2/{+environment}:loadSnapshotloadSnapshotRequest8ComposerProjectsLocationsEnvironmentsLoadSnapshotRequestr   FrP   r\   r^   r6   r5   r_   z>ComposerV1alpha2.ProjectsLocationsEnvironmentsService.<lambda>  r   r6   c                 L    | j                  d      }| j                  |||      S )a"  Migrates an environment to a new major version.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsMigrateRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (Operation) The response message.
      Migrater@   rB   rE   s       r5   r*  z=ComposerV1alpha2.ProjectsLocationsEnvironmentsService.Migrate  r   r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nz\v1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}:migrater   z0composer.projects.locations.environments.migraterK   zv1alpha2/{+name}:migratemigrateEnvironmentRequest3ComposerProjectsLocationsEnvironmentsMigrateRequestr   FrP   r\   r^   r6   r5   r_   z>ComposerV1alpha2.ProjectsLocationsEnvironmentsService.<lambda>  s4    H$:$:pDxH01O&%r6   c                 L    | j                  d      }| j                  |||      S )a  Update an environment.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsPatchRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (Operation) The response message.
      Patchr@   rB   rE   s       r5   r/  z;ComposerV1alpha2.ProjectsLocationsEnvironmentsService.Patch  r   r6   c                  H    t        j                  ddddgdgdgdddd	d
      S )Nr
  PATCHz.composer.projects.locations.environments.patchrK   
updateMaskrL   r   1ComposerProjectsLocationsEnvironmentsPatchRequestr   FrP   r\   r^   r6   r5   r_   z>ComposerV1alpha2.ProjectsLocationsEnvironmentsService.<lambda>*  s6    ("8"8hBxH"^(#M&#r6   c                 L    | j                  d      }| j                  |||      S )aD  Polls Airflow CLI command execution and fetches logs.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsPollAirflowCommandRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (PollAirflowCommandResponse) The response message.
      PollAirflowCommandr@   rB   rE   s       r5   r5  zHComposerV1alpha2.ProjectsLocationsEnvironmentsService.PollAirflowCommand8  2     ##$89f__
'  8 8r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nzgv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}:pollAirflowCommandr   z;composer.projects.locations.environments.pollAirflowCommandr   z*v1alpha2/{+environment}:pollAirflowCommandpollAirflowCommandRequest>ComposerProjectsLocationsEnvironmentsPollAirflowCommandRequestPollAirflowCommandResponseFrP   r\   r^   r6   r5   r_   z>ComposerV1alpha2.ProjectsLocationsEnvironmentsService.<lambda>E  4    x/E/E{O%"OB1Z70r6   c                 L    | j                  d      }| j                  |||      S )a  Restart Airflow web server.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsRestartWebServerRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (Operation) The response message.
      RestartWebServerr@   rB   rE   s       r5   r=  zFComposerV1alpha2.ProjectsLocationsEnvironmentsService.RestartWebServerS  r  r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nzev1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}:restartWebServerr   z9composer.projects.locations.environments.restartWebServerrK   z!v1alpha2/{+name}:restartWebServerrestartWebServerRequest<ComposerProjectsLocationsEnvironmentsRestartWebServerRequestr   FrP   r\   r^   r6   r5   r_   z>ComposerV1alpha2.ProjectsLocationsEnvironmentsService.<lambda>`  s4    X-C-CyMxH9/X&.r6   c                 L    | j                  d      }| j                  |||      S )a  Creates a snapshots of a Cloud Composer environment. As a result of this operation, snapshot of environment's state is stored in a location specified in the SaveSnapshotRequest.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsSaveSnapshotRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (Operation) The response message.
      SaveSnapshotr@   rB   rE   s       r5   rB  zBComposerV1alpha2.ProjectsLocationsEnvironmentsService.SaveSnapshotn  r   r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nzav1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}:saveSnapshotr   z5composer.projects.locations.environments.saveSnapshotr   z$v1alpha2/{+environment}:saveSnapshotsaveSnapshotRequest8ComposerProjectsLocationsEnvironmentsSaveSnapshotRequestr   FrP   r\   r^   r6   r5   r_   z>ComposerV1alpha2.ProjectsLocationsEnvironmentsService.<lambda>{  r   r6   c                 L    | j                  d      }| j                  |||      S )a3  Stops Airflow CLI command execution.

      Args:
        request: (ComposerProjectsLocationsEnvironmentsStopAirflowCommandRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (StopAirflowCommandResponse) The response message.
      StopAirflowCommandr@   rB   rE   s       r5   rG  zHComposerV1alpha2.ProjectsLocationsEnvironmentsService.StopAirflowCommand  r6  r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nzgv1alpha2/projects/{projectsId}/locations/{locationsId}/environments/{environmentsId}:stopAirflowCommandr   z;composer.projects.locations.environments.stopAirflowCommandr   z*v1alpha2/{+environment}:stopAirflowCommandstopAirflowCommandRequest>ComposerProjectsLocationsEnvironmentsStopAirflowCommandRequestStopAirflowCommandResponseFrP   r\   r^   r6   r5   r_   z>ComposerV1alpha2.ProjectsLocationsEnvironmentsService.<lambda>  r;  r6   r:   )rl   rm   rn   ro   rp   r   r   rq   r   r  r   r  r  r?   ra   r   r%  r*  r/  r5  r=  rB  rG  rr   rs   s   @r5   r(   r     s
   I-E
8"L8F8&"8F8+'8-)8C8D8&"8"L8G8E8($8&"8"L8($$r6   r(   c                   >     e Zd ZdZdZ fdZddZd e_         xZS )6ComposerV1alpha2.ProjectsLocationsImageVersionsServicez@Service class for the projects_locations_imageVersions resource.r+   c                 N    t         t        j                  |   |       i | _        y r:   )r   r   r*   r   r;   r<   s     r5   r   z?ComposerV1alpha2.ProjectsLocationsImageVersionsService.__init__  s$    BBDRSYZdr6   c                 L    | j                  d      }| j                  |||      S )a*  List ImageVersions for provided location.

      Args:
        request: (ComposerProjectsLocationsImageVersionsListRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (ListImageVersionsResponse) The response message.
      ra   r@   rB   rE   s       r5   ra   z;ComposerV1alpha2.ProjectsLocationsImageVersionsService.List  rb   r6   c                  J    t        j                  ddddgdgg ddddd	d
      S )NzDv1alpha2/projects/{projectsId}/locations/{locationsId}/imageVersionsrJ   z.composer.projects.locations.imageVersions.listrd   )includePastReleasesrg   rh   z v1alpha2/{+parent}/imageVersionsrM   1ComposerProjectsLocationsImageVersionsListRequestListImageVersionsResponseFrP   r\   r^   r6   r5   r_   z?ComposerV1alpha2.ProjectsLocationsImageVersionsService.<lambda>  s4    !7!7XB zJE8M6"r6   r:   r   rs   s   @r5   r*   rM    s"    J.E
8Dr6   r*   c                        e Zd ZdZdZ fdZddZd e_        ddZd e_        ddZ	d	 e	_        dd
Z
d e
_         xZS )3ComposerV1alpha2.ProjectsLocationsOperationsServicez=Service class for the projects_locations_operations resource.r-   c                 N    t         t        j                  |   |       i | _        y r:   )r   r   r,   r   r;   r<   s     r5   r   z<ComposerV1alpha2.ProjectsLocationsOperationsService.__init__  s$    ??OPVWdr6   c                 L    | j                  d      }| j                  |||      S )a7  Starts asynchronous cancellation on a long-running operation. The server makes a best effort to cancel the operation, but success is not guaranteed. If the server doesn't support this method, it returns `google.rpc.Code.UNIMPLEMENTED`. Clients can use Operations.GetOperation or other methods to check whether the cancellation succeeded or whether the operation completed despite cancellation. On successful cancellation, the operation is not deleted; instead, it becomes an operation with an Operation.error value with a google.rpc.Status.code of `1`, corresponding to `Code.CANCELLED`.

      Args:
        request: (ComposerProjectsLocationsOperationsCancelRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (Empty) The response message.
      Cancelr@   rB   rE   s       r5   rX  z:ComposerV1alpha2.ProjectsLocationsOperationsService.Cancel  r   r6   c                  F    t        j                  ddddgdgg ddddd	
      S )NzWv1alpha2/projects/{projectsId}/locations/{locationsId}/operations/{operationsId}:cancelr   z-composer.projects.locations.operations.cancelrK   zv1alpha2/{+name}:cancelcancelOperationRequest0ComposerProjectsLocationsOperationsCancelRequestr   FrP   r\   r^   r6   r5   r_   z<ComposerV1alpha2.ProjectsLocationsOperationsService.<lambda>  s4    8#9#9kAxH/.L"$r6   c                 L    | j                  d      }| j                  |||      S )a  Deletes a long-running operation. This method indicates that the client is no longer interested in the operation result. It does not cancel the operation. If the server doesn't support this method, it returns `google.rpc.Code.UNIMPLEMENTED`.

      Args:
        request: (ComposerProjectsLocationsOperationsDeleteRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (Empty) The response message.
      r   r@   rB   rE   s       r5   r   z:ComposerV1alpha2.ProjectsLocationsOperationsService.Delete  r   r6   c                  F    t        j                  ddddgdgg ddddd	
      S )NPv1alpha2/projects/{projectsId}/locations/{locationsId}/operations/{operationsId}r   z-composer.projects.locations.operations.deleterK   rL   rM   0ComposerProjectsLocationsOperationsDeleteRequestr   FrP   r\   r^   r6   r5   r_   z<ComposerV1alpha2.ProjectsLocationsOperationsService.<lambda>  s4    8#9#9dAxH(L"$r6   c                 L    | j                  d      }| j                  |||      S )a  Gets the latest state of a long-running operation. Clients can use this method to poll the operation result at intervals as recommended by the API service.

      Args:
        request: (ComposerProjectsLocationsOperationsGetRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (Operation) The response message.
      r?   r@   rB   rE   s       r5   r?   z7ComposerV1alpha2.ProjectsLocationsOperationsService.Get	  rH   r6   c                  F    t        j                  ddddgdgg ddddd	
      S )Nr^  rJ   z*composer.projects.locations.operations.getrK   rL   rM   -ComposerProjectsLocationsOperationsGetRequestr   FrP   r\   r^   r6   r5   r_   z<ComposerV1alpha2.ProjectsLocationsOperationsService.<lambda>  s4     6 6d>xH(I&!r6   c                 L    | j                  d      }| j                  |||      S )a  Lists operations that match the specified filter in the request. If the server doesn't support this method, it returns `UNIMPLEMENTED`.

      Args:
        request: (ComposerProjectsLocationsOperationsListRequest) input message
        global_params: (StandardQueryParameters, default: None) global arguments
      Returns:
        (ListOperationsResponse) The response message.
      ra   r@   rB   rE   s       r5   ra   z8ComposerV1alpha2.ProjectsLocationsOperationsService.List$  rb   r6   c                  J    t        j                  ddddgdgg ddddd	d
      S )NzAv1alpha2/projects/{projectsId}/locations/{locationsId}/operationsrJ   z+composer.projects.locations.operations.listrK   re   zv1alpha2/{+name}/operationsrM   .ComposerProjectsLocationsOperationsListRequestListOperationsResponseFrP   r\   r^   r6   r5   r_   z<ComposerV1alpha2.ProjectsLocationsOperationsService.<lambda>1  s4    !7!7U?xH83J3"r6   r:   )rl   rm   rn   ro   rp   r   rX  rq   r   r?   ra   rr   rs   s   @r5   r,   rU    sO    G+E
8F8F8C8Dr6   r,   c                   &     e Zd ZdZdZ fdZ xZS ))ComposerV1alpha2.ProjectsLocationsServicez2Service class for the projects_locations resource.r/   c                 N    t         t        j                  |   |       i | _        y r:   )r   r   r.   r   r;   r<   s     r5   r   z2ComposerV1alpha2.ProjectsLocationsService.__init__D  s#    55tEfMdr6   rl   rm   rn   ro   rp   r   rr   rs   s   @r5   r.   rh  ?  s    < E r6   r.   c                   &     e Zd ZdZdZ fdZ xZS ) ComposerV1alpha2.ProjectsServicez(Service class for the projects resource.r1   c                 N    t         t        j                  |   |       i | _        y r:   )r   r   r0   r   r;   r<   s     r5   r   z)ComposerV1alpha2.ProjectsService.__init__N  s#    ,,d<VDdr6   rj  rs   s   @r5   r0   rl  I  s    2E r6   r0   )rM   NTNNFFNNNN)!rl   rm   rn   ro   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$   r&   r(   r*   r,   r.   r0   rr   rs   s   @r5   r   r   
   sK   G//(9-(=>'(*"."+),()-6:/4<@?C	/8>XE\E\ >@>8O8O >@#h6M6M #JE1H1H ENOHD[D[ ObOAXAX Ob#h6M6M #JxX-D-D xt#h.E.E #Jt8+B+B tl!8!8 //  r6   r   N)
ro   
__future__r   apitools.base.pyr   7googlecloudsdk.generated_clients.apis.composer.v1alpha2r   rn  BaseApiClientr   r^   r6   r5   <module>r     s'    = ' % jGx-- Gr6   