
                             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                  e	j                         Zy)(Add a Hive job to the workflow template.    )absolute_import)division)unicode_literals)dataproc)base)workflow_templates)hiveEXAMPLESa        To add a Hive job executing query 'QUERY' to a the workflow template
      'my-workflow-template' in region 'us-central1' with step-id 'my-step-id'
      , run:

        $ {command} --step-id=my-step_id -e=QUERY --workflow-template=my-workflow-template --region=us-central1
      c                   0    e Zd ZdZeZed        Zd Zd Z	y)Hiver   c                     t         j                  j                  |       t        j                  | j                               }t        j                  ||j                         y N)	r
   HiveBaseArgsdpDataprocReleaseTrackr	   AddWorkflowTemplatesArgsapi_version)clsparserr   s      7lib/surface/dataproc/workflow_templates/add_job/hive.pyr   z	Hive.Args+   sA    MMv{{3++-.H//8L8LM    c                 v    t         j                  j                  ||||       t        j                  |||       y r   )r
   r   ConfigureJobr	   ConfigureOrderedJob)selfmessagesjobfiles_by_typeargss        r   r   zHive.ConfigureJob1   s,    MMxmTB**8S$?r   c                    | j                  |       t        j                  | j                               }t	        j
                  ||      }| j                  |j                  || j                  |       t	        j                  |||      S r   )
PopulateFilesByTyper   r   r   r	    CreateWorkflowTemplateOrderedJobr   r   r!   AddJobToWorkflowTemplate)r   r"   r   ordered_jobs       r   RunzHive.Run5   sx    T"{{4,,./H$EEhKh''d6H6H$O66tX7BD Dr   N)
__name__
__module____qualname____doc__DETAILED_HELPdetailed_helpclassmethodr   r   r(    r   r   r   r   &   s*    0-N N
@Dr   r   N)r,   
__future__r   r   r   googlecloudsdk.api_lib.dataprocr   r   googlecloudsdk.callioper   #googlecloudsdk.command_lib.dataprocr	   (googlecloudsdk.command_lib.dataproc.jobsr
   r-   r   Commandr   r0   r   r   <module>r7      sH    / &  ' : ( B 9 	
	D4==$,, Dr   