NdZddlmZddlmZddlmZddlZddlmZddlm Z ddlm Z dd lm Z Gd d ejZ y) zCCommand to export files into a Cloud Composer environment's bucket.)absolute_import)division)unicode_literalsN)base)flags) resource_args) storage_utilc*eZdZdZdZedZdZy)ExportaZExport DAGs from an environment into local storage or Cloud Storage. If the SOURCE is a directory, it and its contents are are exported recursively. If no SOURCE is provided, the entire contents of the environment's DAGs directory will be exported. Colliding files in the DESTINATION will be overwritten. If a file exists in the DESTINATION but there is no corresponding file to overwrite it, it is untouched. ## EXAMPLES Suppose the environment `myenv`'s Cloud Storage bucket has the following structure: gs://the-bucket | +-- dags | | | +-- file1.py | +-- file2.py | | | +-- subdir1 | | | | | +-- file3.py | | +-- file4.py And the local directory '/foo' has the following structure: /foo | +-- file1.py +-- fileX.py | | | +-- subdir1 | | | | | +-- file3.py | | +-- fileY.py The following command: {command} myenv --destination=/foo would result in the following structure in the local '/foo' directory: /foo | +-- file1.py +-- file2.py +-- fileX.py | | | +-- subdir1 | | | | | +-- file3.py | | +-- file4.py | | +-- fileY.py The local files '/foo/file1.py' and '/foo/subdir1/file3.py' will be overwritten with the contents of the corresponding files in the Cloud Storage bucket. If instead we had run {command} myenv --source=subdir1/file3.py --destination=/foo the resulting local directory structure would be the following: /foo | +-- file1.py +-- file3.py +-- fileX.py | | | +-- subdir1 | | | | | +-- file3.py | | +-- fileY.py No local files would be overwritten since 'gs://the-bucket/dags/subdir1/file3.py' was written to '/foo/file3.py' instead of 'foo/subdir1/file3.py'. dagsctj|ddtj|tj tj |y)Nz.from whose Cloud Storage bucket to export DAGsF) positional)rAddEnvironmentResourceArgrAddExportSourceFlagr SUBDIR_BASEAddExportDestinationFlag)parsers 8lib/surface/composer/environments/storage/dags/export.pyArgsz Export.Argsqs?++@ ff&8&89 ""6*ctj|jd|jjj }t jtj|jxsdjt j}tj|||j|jS)Nz--source*) release_track)r WarnIfWildcardIsPresentsourceCONCEPTS environmentParse posixpathjoinr rstripsep destination ReleaseTrack)selfargsenv_ref source_paths rRunz Export.Runys((jAmm''--/G..!3!3"&++"4!;!;IMM!JLK    '') ++rN)__name__ __module__ __qualname____doc__r staticmethodrr)rrr r s'Ob++++rr )r- __future__rrrrgooglecloudsdk.callioper#googlecloudsdk.command_lib.composerrrr Commandr r/rrr4s5J&'(5=<d+T\\d+r