hdZddlmZddlmZddlmZddlZddlmZddlm 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)ExportasExport data 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 data 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 | +-- data | | | +-- file1.txt | +-- file2.txt | | | +-- subdir1 | | | | | +-- file3.txt | | +-- file4.txt And the local directory '/foo' has the following structure: /foo | +-- file1.txt +-- fileX.txt | | | +-- subdir1 | | | | | +-- file3.txt | | +-- fileY.txt The following command: {command} myenv --destination=/foo would result in the following structure in the local '/foo' directory: /foo | +-- file1.txt +-- file2.txt +-- fileX.txt | | | +-- subdir1 | | | | | +-- file3.txt | | +-- file4.txt | | +-- fileY.txt The local files '/foo/file1.txt' and '/foo/subdir1/file3.txt' 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.txt --destination=/foo the resulting local directory structure would be the following: /foo | +-- file1.txt +-- file3.txt +-- fileX.txt | | | +-- subdir1 | | | | | +-- file3.txt | | +-- fileY.txt No local files would be overwritten since 'gs://the-bucket/dags/subdir1/file3.txt' was written to '/foo/file3.txt' instead of 'foo/subdir1/file3.txt'. datactj|ddtj|tj tj |y)Nz/from whose Cloud Storage bucket to export data.F) positional)rAddEnvironmentResourceArgrAddExportSourceFlagr SUBDIR_BASEAddExportDestinationFlag)parsers 8lib/surface/composer/environments/storage/data/export.pyArgsz Export.Argsqs?++A 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