src/tools/screenshot_testing/update_golden_screenshots.py

   1 # Copyright 2014 The Chromium Authors. All rights reserved.
   2 # Use of this source code is governed by a BSD-style license that can be
   3 # found in the LICENSE file.
   4
   5
   6 import sys
   7 import getopt
   8 import os
   9
  10 here = os.path.realpath(__file__)
  11 src_path = (os.path.normpath(os.path.join(here, '..', '..', '..')))
  12 sys.path.append(os.path.normpath(os.path.join(src_path, '..', 'depot_tools')))
  13
  14 USAGE = 'The utility uploads .png files to ' \
  15         'chrome-os-oobe-ui-screenshot-testing Google Storage bucket.\n' \
  16         '-i:\n\tdirectory with .png files which have to be uploaded\n' \
  17         '-o (optional):\n\tdirectory to store generated .sha1 files. ' \
  18         'Is set to chrome/browser/chromeos/login/screenshot_testing' \
  19         '/golden_screenshots by default\n--help:\n\thelp'
  20
  21
  22 import upload_to_google_storage
  23 import download_from_google_storage
  24
  25 def upload(png_path):
  26
  27   # Creating a list of files which need to be uploaded to Google Storage:
  28   # all .png files from the directory containing golden screenshots.
  29   target = []
  30   for file in os.listdir(png_path):
  31     if file.endswith('.png'):
  32       target.append(os.path.join(png_path, file))
  33
  34   # Creating a standard gsutil object, assuming there are depot_tools
  35   # and everything related is set up already.
  36   gsutil_path = os.path.abspath(os.path.join(src_path, '..', 'depot_tools',
  37                                              'third_party', 'gsutil',
  38                                              'gsutil'))
  39   gsutil = download_from_google_storage.Gsutil(gsutil_path,
  40                                                boto_path=None,
  41                                                bypass_prodaccess=True)
  42
  43   # URL of the bucket used for storing screenshots.
  44   bucket_url = 'gs://chrome-os-oobe-ui-screenshot-testing'
  45
  46   # Uploading using the most simple way,
  47   # see depot_tools/upload_to_google_storage.py to have better understanding
  48   # of this False and 1 arguments.
  49   upload_to_google_storage.upload_to_google_storage(target, bucket_url, gsutil,
  50                                                     False, False, 1, False)
  51
  52   print 'All images are uploaded to Google Storage.'
  53
  54 def move_sha1(from_path, to_path):
  55   from shutil import move
  56   for file in os.listdir(from_path):
  57     if (file.endswith('.sha1')):
  58       old_place = os.path.join(from_path, file)
  59       new_place = os.path.join(to_path, file)
  60       if not os.path.exists(os.path.dirname(new_place)):
  61         os.makedirs(os.path.dirname(new_place))
  62       move(old_place, new_place)
  63
  64 def main(argv):
  65   png_path = ''
  66   sha1_path = os.path.join(src_path,
  67                            'chrome', 'browser', 'chromeos', 'login',
  68                            'screenshot_testing', 'golden_screenshots')
  69   try:
  70     opts, args = getopt.getopt(argv,'i:o:', ['--help'])
  71   except getopt.GetoptError:
  72     print USAGE
  73     sys.exit(1)
  74   for opt, arg in opts:
  75     if opt == '--help':
  76       print USAGE
  77       sys.exit()
  78     elif opt == '-i':
  79       png_path = arg
  80     elif opt =='-o':
  81       sha1_path = arg
  82
  83   if png_path == '':
  84     print USAGE
  85     sys.exit(1)
  86
  87   png_path = os.path.abspath(png_path)
  88   sha1_path = os.path.abspath(sha1_path)
  89
  90   upload(png_path)
  91   move_sha1(png_path, sha1_path)
  92
  93   # TODO(elizavetai): Can this git stuff be done automatically?
  94   print 'Please add new .sha1 files from ' \
  95         + str(sha1_path) + \
  96         ' to git manually.'
  97
  98 if __name__ == "__main__":
  99  main(sys.argv[1:])