
                             S r SSKJr  SSKJr  SSKJr  SSKJr  SSKJ	r	  SSK
Jr  SSK
Jr  SS	KJr  SS
KrSS0r " S S\	R"                  5      rg
)z-Set DAG timeout on workflow template command.    )absolute_import)division)unicode_literals)dataproc)base)flags)workflow_templates)logNEXAMPLESz      To add a DAG timeout of 2h (or update an existing one to 2h) on a workflow
      template named ``my-workflow-template'' in region ``us-central1'', run:

        $ {command} my-workflow-template --region=us-central1 --dag-timeout=2h"
      c                   2    \ rS rSrSr\r\S 5       rS r	Sr
g)SetDagTimeout'   z'Set DAG timeout on a workflow template.c                     [         R                  " U R                  5       5      n[        R                  " US5        [
        R                  " USUR                  5        g )NTzset the DAG timeout on)dpDataprocReleaseTrackr	   AddDagTimeoutFlagr   AddTemplateResourceArgapi_version)clsparserr   s      :lib/surface/dataproc/workflow_templates/set_dag_timeout.pyArgsSetDagTimeout.Args,   sE    {{3++-.H((6	  )A!)!5!57    c                    [         R                  " U R                  5       5      nUR                  R                  R                  5       nUR                  X1R                  5      n[        R                  " UR                  5      S-   Ul        UR                  R                  R                  U5      n[        R                   R#                  SR%                  UR                  UR'                  5       5      5        U$ )Nsz Set a DAG timeout of {0} on {1}.)r   r   r   CONCEPTStemplateParseGetRegionsWorkflowTemplateversionsix	text_typedag_timeout
dagTimeoutclient"projects_regions_workflowTemplatesUpdater
   statusPrintformatName)selfargsr   template_refworkflow_templateresponses         r   RunSetDagTimeout.Run3   s    {{4,,./H==))//1L ;;ll$ $'==1A1A#BS#H AAHHH JJ7>>$$l&7&7&9; <Or    N)__name__
__module____qualname____firstlineno____doc__DETAILED_HELPdetailed_helpclassmethodr   r3   __static_attributes__r5   r   r   r   r   '   s!    /-7 7r   r   )r:   
__future__r   r   r   googlecloudsdk.api_lib.dataprocr   r   googlecloudsdk.callioper   #googlecloudsdk.command_lib.dataprocr   r	   googlecloudsdk.corer
   r#   r;   CreateCommandr   r5   r   r   <module>rE      sH    4 &  ' : ( 5 B # 
 	
D&& r   