Chromium Code Reviews| Index: build/android/pylib/utils/google_storage_helper.py |
| diff --git a/build/android/pylib/utils/google_storage_helper.py b/build/android/pylib/utils/google_storage_helper.py |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..2fbbc1513ebc5aa8ea9701b9edc3ab63645c4b75 |
| --- /dev/null |
| +++ b/build/android/pylib/utils/google_storage_helper.py |
| @@ -0,0 +1,57 @@ |
| +# Copyright 2017 The Chromium Authors. All rights reserved. |
|
the real yoland
2017/02/24 23:00:31
hmm, why not using upload_to_google_storage.py?
mikecase (-- gone --)
2017/02/24 23:41:30
Since its very specific. It requires you to provid
|
| +# Use of this source code is governed by a BSD-style license that can be |
| +# found in the LICENSE file. |
| + |
| +"""Helper functions to upload data to Google Storage. |
| + |
| +Text data should be streamed to logdog using |logdog_helper| module. |
| +Due to logdog not having image or HTML viewer, those instead should be uploaded |
| +to Google Storage directly using this module. |
| +""" |
| + |
| +import logging |
| +import os |
| +import sys |
| +import time |
| + |
| +from devil.utils import cmd_helper |
| +from pylib.constants import host_paths |
| +from pylib.utils import decorators |
| + |
| +sys.path.append(os.path.join(host_paths.DIR_SOURCE_ROOT, 'build')) |
| +import find_depot_tools # pylint: disable=import-error |
| + |
| +_URL_TEMPLATE = 'https://storage.googleapis.com/%s/' |
| + |
| + |
| +@decorators.NoRaiseException(default_return_value='') |
| +def upload(name, filepath, bucket): |
| + """Uploads data to Google Storage. |
| + |
| + Args: |
| + name: Name of the file on Google Storage. |
| + filepath: Path to file you want to upload. |
| + bucket: Bucket to upload file to. |
| + """ |
| + gs_path = os.path.join('gs://%s/' % bucket, name) |
| + logging.info('Uploading %s to %s', filepath, gs_path) |
| + cmd_helper.RunCmd( |
| + [os.path.join(find_depot_tools.DEPOT_TOOLS_PATH, 'gsutil.py'), 'cp', |
| + filepath, gs_path]) |
| + |
| + return os.path.join(_URL_TEMPLATE % bucket, name) |
| + |
| + |
| +def unique_name(basename, timestamp=True, device=None): |
| + """Helper function for creating a unique name for a logdog stream. |
| + |
| + Args: |
| + basename: Base of the unique name. |
| + timestamp: Whether or not to add a timestamp to name. |
| + device: Device to add device serial of to name. |
| + """ |
| + return '%s%s%s' % ( |
| + basename, |
| + '_%s' % time.strftime('%Y%m%dT%H%M%S', time.localtime()) |
| + if timestamp else '', |
| + '_%s' % device.serial if device else '') |