
                             d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlm	Z	 ddl
mZ ddlmZ d	d
iZ G d de	j                        Zy)z2Remove DAG timeout from workflow template command.    )absolute_import)division)unicode_literals)dataproc)base)flags)logEXAMPLESz      To remove a DAG timeout from a workflow template named
      ``my-workflow-template'' in region ``us-central1'', run:

        $ {command} my-workflow-template --region=us-central1"
      c                   *    e Zd ZdZeZed        Zd Zy)RemoveDagTimeoutz,Remove DAG timeout from a workflow template.c                     t        j                  | j                               }t        j                  |d|j
                         y )Nzremove the DAG timeout from)dpDataprocReleaseTrackr   AddTemplateResourceArgapi_version)clsparserr   s      =lib/surface/dataproc/workflow_templates/remove_dag_timeout.pyArgszRemoveDagTimeout.Args*   s5    {{3++-.H	  )F!)!5!57    c                    t        j                  | j                               }|j                  j                  j                         }|j                  ||j                        }d |_        |j                  j                  j                  |      }t        j                  j                  dj                  |j!                                      |S )NzRemoved DAG timeout from {0}.)r   r   r   CONCEPTStemplateParseGetRegionsWorkflowTemplateversion
dagTimeoutclient"projects_regions_workflowTemplatesUpdater	   statusPrintformatName)selfargsr   template_refworkflow_templateresponses         r   RunzRemoveDagTimeout.Run0   s    {{4,,./H==))//1L ;;dll$ $( AAHHH JJ4;; Or   N)	__name__
__module____qualname____doc__DETAILED_HELPdetailed_helpclassmethodr   r+    r   r   r   r   %   s!    4-7 7
r   r   N)r/   
__future__r   r   r   googlecloudsdk.api_lib.dataprocr   r   googlecloudsdk.callioper   #googlecloudsdk.command_lib.dataprocr   googlecloudsdk.corer	   r0   CreateCommandr   r3   r   r   <module>r:      sB    9 &  ' : ( 5 # 	
t)) r   