J `dZddlmZddlmZddlmZddlmZddlmZGddeZ d Z y ) #Factory class for RayBatch message.)absolute_import)division)unicode_literals)flags)local_file_uploaderceZdZdZdZdZy)RayBatchFactoryrc||_y)zXFactory class for RayBatch message. Args: dataproc: A Dataproc instance. N)dataproc)selfr s Dlib/googlecloudsdk/command_lib/dataproc/batches/ray_batch_factory.py__init__zRayBatchFactory.__init__s DMc|i}|jr|j|d<i}|jg|d<tj|r7|js t dtj |j|}|dd|d<|j||jjjdi|S)aupload user local files and creates a RayBatch message. Upload user local files and point URIs to the local files to the uploaded URIs. Creates a RayBatch message from parsed arguments. Args: args: Parsed arguments. Returns: RayBatch: A RayBatch message. Raises: AttributeError: Bucket is required to upload local files, but not specified. argsmainPythonFileUriz --deps-bucket was not specified.r) rMAIN_PYTHON_FILEr HasLocalFiles deps_bucketAttributeErrorUploadupdater messagesRayBatch)r rkwargs dependenciess rUploadLocalFilesAndGetMessagez-RayBatchFactory.UploadLocalFilesAndGetMessage%s"F yyyyfVnL*.)>)>(?L$%((6   ?@@(//0@0@,Ol)55H(I!(LL$% MM, *4== ! ! * * 4V 44rN)__name__ __module__ __qualname____doc__rrrrrr r s+'5rr ctj|tj|tj|y)z+Adds arguments related to RayBatch message.N)rAddMainPythonFileAddArgs AddBucket)parsers r AddArgumentsr)Os*&!--//&rN) r# __future__rrr#googlecloudsdk.command_lib.dataprocrrobjectr r)rrrr-s+ *&'5C25f25jr