
                             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
\R                  5      rg)zBase class for Pig Job.    )absolute_import)division)unicode_literals)encoding)arg_parsers)base)utilc                   H    \ rS rSrSr\S 5       r\S 5       r\S 5       rSr	g)PigBase   zSubmit a Pig job to a cluster.c           
         U R                  SS9nUR                  SSSSS/ SS	9  UR                  S
SSS9  U R                  S[        R                  " 5       S/ SS9  U R                  S[        R                  " 5       SSS9  U R                  S[        R                  " 5       SSS9  U R                  S[
        R                  S9  U R                  SSSS9  U R                  S[        R                  " 5       SS S9  g!)"z7Performs command-line argument parsing specific to Pig.T)requiredz	--executez-eQUERYqueriesappendz*A Pig query to execute as part of the job.)metavardestactiondefaulthelpz--filez-fz=HCFS URI of file containing Pig script to execute as the job.)r   z--jarsJARzQComma separated list of jar files to be provided to Pig and MR. May contain UDFs.)typer   r   r   z--paramszPARAM=VALUEz>A list of key value pairs to set variables in the Pig queries.)r   r   r   z--propertieszPROPERTY=VALUEz+A list of key value pairs to configure Pig.z--properties-filez--continue-on-failure
store_truez,Whether to continue if a single query fails.)r   r   z--driver-log-levelszPACKAGE=LEVELzpA list of package to log4j log level pairs to configure driver logging. For example: root=FATAL,com.example=INFON)add_mutually_exclusive_groupadd_argumentr   ArgListArgDictjob_utilPROPERTIES_FILE_HELP_TEXT)parserdrivers     3lib/googlecloudsdk/command_lib/dataproc/jobs/pig.pyArgsPigBase.Args    sM    00$0?F
T9  ; $L  N   ".  0   "M	  O
   " :	  <
 //  1 ;  =   "B	  D    c                 4    U R                   U R                  S.$ )Njarsfiler'   )argss    r"   GetFilesByTypePigBase.GetFilesByTypeM   s     				 r%   c                    U R                  UR                  US   US   US9nUR                  (       a  U R                  UR                  S9Ul        UR
                  (       a:  [        R                  " UR
                  U R                   R                  5      Ul	        [        R                  " UR                  UR                  5      nU(       a.  [        R                  " X`R                   R                  SS9Ul        XQl        g)z-Populates the pigJob member of the given job.r(   r)   )continueOnFailurejarFileUrisqueryFileUriloggingConfig)r   T)
sort_itemsN)PigJobcontinue_on_failurer   	QueryList	queryListparamsr   DictToAdditionalPropertyMessageScriptVariablesValuescriptVariablesr   BuildJobProperties
propertiesproperties_filePropertiesValuepigJob)messagesjobfiles_by_typelogging_configr*   pig_jobjob_propertiess          r"   ConfigureJobPigBase.ConfigureJobS   s     oo22!&)"6*$	  &G ||",,T\\,Bg{{ ( H H
++x;;!=g 00--/N#CC
//99dLg Jr%    N)
__name__
__module____qualname____firstlineno____doc__staticmethodr#   r+   rF   __static_attributes__rH   r%   r"   r   r      sB    &*D *DX  
  r%   r   N)rM   
__future__r   r   r   apitools.base.pyr   googlecloudsdk.callioper   (googlecloudsdk.command_lib.dataproc.jobsr   job_baser	   r   JobBaser   rH   r%   r"   <module>rV      s3      &  ' % / E EMh Mr%   