
    ;	                         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 Trino job to the workflow template.    )absolute_import)division)unicode_literals)dataproc)base)workflow_templates)trinoEXAMPLESa        To add a Trino job that executes 'QUERY' to the workflow template
      'my-workflow-template' in the 'us-central1' region with '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)Trinor   c                     t         j                  j                  |       t        j                  | j                               }t        j                  ||j                         y N)	r
   	TrinoBaseArgsdpDataprocReleaseTrackr	   AddWorkflowTemplatesArgsapi_version)clsparserr   s      8lib/surface/dataproc/workflow_templates/add_job/trino.pyr   z
Trino.Args,   sA    	OO {{3++-.H//8L8LM    c           	          t         j                  j                  |||| j                  ||j                        |       t        j                  |||       y r   )r
   r   ConfigureJobBuildLoggingConfigdriver_log_levelsr	   ConfigureOrderedJob)selfmessagesjobfiles_by_typeargss        r   r   zTrino.ConfigureJob2   sH    	OO  #}$*@*@A4I **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	Trino.Run8   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*    1-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   r2   r   r   <module>r9      sH    0 &  ' : ( B : 	

DEOOT\\ Dr   