
    
                         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	lmZ d
diZ G d de	j"                        Zy)z*Remove Job from workflow template command.    )absolute_import)division)unicode_literals)dataproc)base)flags)log)
console_ioEXAMPLESz      To remove a job with step ID 'step-id' from a workflow template
      'workflow-template' in region 'us-central1', run:

        $ {command} workflow-template --region=us-central1 --step-id=step-id
      c                   *    e Zd ZdZeZed        Zd Zy)	RemoveJobz$Remove a job from workflow template.c                     t        j                  | j                               }|j                  ddt        d       t        j                  |d|j                         y )Nz	--step-idSTEP_IDz:The step ID of the job in the workflow template to remove.)metavartypehelpz
remove job)api_version)dpDataprocReleaseTrackadd_argumentstrr   AddTemplateResourceArgr   )clsparserr   s      5lib/surface/dataproc/workflow_templates/remove_job.pyArgszRemoveJob.Args+   sW    {{3++-.H
I	  K
 
  (*>*>@    c                    t        j                  | j                               }|j                  j                  j                         }|j                  ||j                        }|j                  }d}g }|D ]j  }|j                  |j                  k7  r|j                  |       .t        j                  dj                  |j                  |j                        d       d}l |s:t!        j"                  dj                  |j                  |j                               y ||_        |j$                  j&                  j)                  |      }	|	S )NFz;The job [{0}] will be removed from workflow template [{1}].T)messagecancel_on_noz6Step id [{0}] is not found in workflow template [{1}].)r   r   r   CONCEPTStemplateParseGetRegionsWorkflowTemplateversionjobsstepIdstep_idappendr
   PromptContinueformatidr	   errorclient"projects_regions_workflowTemplatesUpdate)
selfargsr   template_refworkflow_templater'   job_removednew_jobsordered_jobresponses
             r   RunzRemoveJob.Run6   s   {{4,,./H==))//1L ;;dll$ !!DKH			t||	+$!!%vdll4E4H4HI	   	iiHOO
,,),,. /%AAHHHOr   N)	__name__
__module____qualname____doc__DETAILED_HELPdetailed_helpclassmethodr   r:    r   r   r   r   &   s#    ,-@ @r   r   N)r>   
__future__r   r   r   googlecloudsdk.api_lib.dataprocr   r   googlecloudsdk.callioper   #googlecloudsdk.command_lib.dataprocr   googlecloudsdk.corer	   googlecloudsdk.core.consoler
   r?   UpdateCommandr   rB   r   r   <module>rJ      sE    1 &  ' : ( 5 # 2 	
."" .r   