| """Collection of utilities to work with Google Cloud Storage and gsutil.""" |
| |
| import os |
| |
| |
| def gsutil(): |
| """Returns the path to the gsutil binary.""" |
| # Search the normal install path for gsutil first. This should succeed |
| # most of the time. (A notable exception being GCE instances, which |
| # have gsutil on the PATH somewhere else.) |
| gsutil_path = os.path.expanduser('~/gsutil/gsutil') |
| if os.path.exists(gsutil_path): |
| return gsutil_path |
| |
| # Otherwise, assume that gsutil is on the PATH so that this will find it. |
| return 'gsutil' |
| |
| |
| def upload_directory_cmd(local_dir, gcs_dir): |
| """Command to upload the contents of the local directory to the GCS bucket. |
| |
| Note that this assumes a gsutil install located at ~/gsutil. |
| |
| Args: |
| local_dir: local directory to upload the contents of (e.g. path/to/foo) |
| gcs_dir: directory on GCS to upload to (e.g. gs://my-bucket/foo) |
| Returns: |
| The command (as a list) to execute to upload a directory to GCS. |
| """ |
| return [gsutil(), 'cp', '-r', local_dir, gcs_dir] |
