Upload SKPs to chrome-partner-telemetry

tools/skp/webpages_playback.py

 #!/usr/bin/env python
 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """Archives or replays webpages and creates SKPs in a Google Storage location.
 
 To archive webpages and store SKP files (archives should be rarely updated):
 
 cd skia
@@ skipping 53 matching lines @@
 import traceback
 
 sys.path.insert(0, os.getcwd())
 
 from common.py.utils import gs_utils
 from common.py.utils import shell_utils
 
 ROOT_PLAYBACK_DIR_NAME = 'playback'
 SKPICTURES_DIR_NAME = 'skps'
 
+PARTNERS_GS_BUCKET = 'gs://chrome-partner-telemetry'
 
 # Local archive and SKP directories.
 LOCAL_PLAYBACK_ROOT_DIR = os.path.join(
     tempfile.gettempdir(), ROOT_PLAYBACK_DIR_NAME)
 LOCAL_REPLAY_WEBPAGES_ARCHIVE_DIR = os.path.join(
     os.path.abspath(os.path.dirname(__file__)), 'page_sets', 'data')
 TMP_SKP_DIR = tempfile.mkdtemp()
 
 # Location of the credentials.json file and the string that represents missing
 # passwords.
@@ skipping 67 matching lines @@
     self._record = parse_options.record
     self._skia_tools = parse_options.skia_tools
     self._non_interactive = parse_options.non_interactive
     self._upload = parse_options.upload
     self._skp_prefix = parse_options.skp_prefix
     data_store_location = parse_options.data_store
     if data_store_location.startswith(gs_utils.GS_PREFIX):
       self.gs = GoogleStorageDataStore(data_store_location)
     else:
       self.gs = LocalFileSystemDataStore(data_store_location)
+    self._upload_to_partner_bucket = parse_options.upload_to_partner_bucket
     self._alternate_upload_dir = parse_options.alternate_upload_dir
     self._telemetry_binaries_dir = os.path.join(parse_options.chrome_src_path,
                                                 'tools', 'perf')
 
     self._local_skp_dir = os.path.join(
         parse_options.output_dir, ROOT_PLAYBACK_DIR_NAME, SKPICTURES_DIR_NAME)
     self._local_record_webpages_archive_dir = os.path.join(
         parse_options.output_dir, ROOT_PLAYBACK_DIR_NAME, 'webpages_archive')
 
     # List of SKP files generated by this script.
@@ skipping 190 matching lines @@
 
       self.gs.upload_dir_contents(
           LOCAL_PLAYBACK_ROOT_DIR, dest_dir=dest_dir_name,
           upload_if=gs_utils.GSUtils.UploadIf.IF_MODIFIED,
           predefined_acl=GS_PREDEFINED_ACL,
           fine_grained_acl_list=GS_FINE_GRAINED_ACL_LIST)
 
       print '\n\n=======New SKPs have been uploaded to %s =======\n\n' % (
           posixpath.join(self.gs.target_name(), dest_dir_name,
                          SKPICTURES_DIR_NAME))
+
+      if self._upload_to_partner_bucket:
+        print '\n\n=======Uploading to Partner bucket %s =======\n\n' % (
+            PARTNERS_GS_BUCKET)
+        partner_gs = GoogleStorageDataStore(PARTNERS_GS_BUCKET)
+        partner_gs.upload_dir_contents(
+            os.path.join(LOCAL_PLAYBACK_ROOT_DIR, SKPICTURES_DIR_NAME),
+            dest_dir=posixpath.join(SKPICTURES_DIR_NAME, dest_dir_name),
+            upload_if=gs_utils.GSUtils.UploadIf.IF_MODIFIED)
+        print '\n\n=======New SKPs have been uploaded to %s =======\n\n' % (
+            posixpath.join(partner_gs.target_name(), SKPICTURES_DIR_NAME,
+                           dest_dir_name))
     else:
       print '\n\n=======Not Uploading to %s=======\n\n' % self.gs.target_type()
       print 'Generated resources are available in %s\n\n' % (
           LOCAL_PLAYBACK_ROOT_DIR)
 
     return 0
 
   def _GetSkiaSkpFileName(self, page_set):
     """Returns the SKP file name for Skia page sets."""
     # /path/to/skia_yahooanswers_desktop.py -> skia_yahooanswers_desktop.py
@@ skipping 146 matching lines @@
             '--non-interactive then the debugger is also run at the end. Debug '
             'builds are recommended because they seem to catch more failures '
             'than Release builds.'),
       default=None)
   option_parser.add_option(
       '', '--upload', action='store_true',
       help=('Uploads to Google Storage or copies to local filesystem storage '
             ' if this is True.'),
       default=False)
   option_parser.add_option(
+      '', '--upload_to_partner_bucket', action='store_true',
+      help=('Uploads SKPs to the chrome-partner-telemetry Google Storage '
+            'bucket if true.'),
+      default=False)
+  option_parser.add_option(
       '', '--data_store',
     help=('The location of the file storage to use to download and upload '
           'files. Can be \'gs://<bucket>\' for Google Storage, or '
           'a directory for local filesystem storage'),
       default='gs://chromium-skia-gm')
   option_parser.add_option(
       '', '--alternate_upload_dir',
       help= ('Uploads to a different directory in Google Storage or local '
              'storage if this flag is specified'),
       default=None)
@@ skipping 20 matching lines @@
            '--skia_tools is specified then the debugger is not run.',
       default=False)
   option_parser.add_option(
       '', '--skp_prefix',
       help='Prefix to add to the names of generated SKPs.',
       default=None)
   options, unused_args = option_parser.parse_args()
 
   playback = SkPicturePlayback(options)
   sys.exit(playback.Run())