
    +
                     `    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\5      r	S	 r
g
)&Factory class for SparkRBatch message.    )absolute_import)division)unicode_literals)flags)local_file_uploaderc                   $    \ rS rSrSrS rS rSrg)SparkRBatchFactory   r   c                     Xl         g)zOFactory class for SparkRBatch message.

Args:
  dataproc: A Dataproc instance.
Ndataproc)selfr   s     Glib/googlecloudsdk/command_lib/dataproc/batches/sparkr_batch_factory.py__init__SparkRBatchFactory.__init__   s	     M    c                    0 nUR                   (       a  UR                   US'   0 nUR                  /US'   UR                  (       a  UR                  US'   UR                  (       a  UR                  US'   [        R
                  " U5      (       a=  UR                  (       d  [        S5      e[        R                  " UR                  U5      nUS   S   US'   UR                  U5        U R                  R                  R                  " S0 UD6$ )a`  Upload local files and creates a SparkRBatch message.

Upload user local files and change local file URIs to point to the uploaded
URIs.
Creates a SparkRBatch message based on parsed arguments.

Args:
  args: Parsed arguments.

Returns:
  A SparkRBatch message.

Raises:
  AttributeError: Bucket is required to upload local files, but not
  specified.
argsmainRFileUrifileUrisarchiveUrisz --deps-bucket was not specified.r    )r   MAIN_R_FILEfilesarchivesr   HasLocalFilesdeps_bucketAttributeErrorUploadupdater   messagesSparkRBatch)r   r   kwargsdependenciess       r   UploadLocalFilesAndGetMessage0SparkRBatchFactory.UploadLocalFilesAndGetMessage%   s    " FyyyyfVnL %)$4$4#5L zz!%l:}}$(MMl=!((66?@@(//0@0@,Ol $0#?#BL  MM,==!!--777r   r   N)__name__
__module____qualname____firstlineno____doc__r   r&   __static_attributes__r   r   r   r
   r
      s    .,8r   r
   c                     [         R                  " U 5        [         R                  " U 5        [         R                  " U 5        [         R                  " U 5        [         R
                  " U 5        g )N)r   AddMainRFileAddArgsAddOtherFilesAddArchives	AddBucket)parsers    r   AddArgumentsr5   T   sD    V--fF//&r   N)r,   
__future__r   r   r   #googlecloudsdk.command_lib.dataprocr   r   objectr
   r5   r   r   r   <module>r9      s+     - &  ' 5 C78 78tr   