OLD | NEW |
(Empty) | |
| 1 #!/usr/bin/env python |
| 2 # Copyright 2014 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. |
| 5 |
| 6 """Script to download sdk/extras packages on the bots from google storage. |
| 7 |
| 8 The script expects arguments that specify zips file in the google storage |
| 9 bucket named: <dir in SDK extras>_<package name>_<version>.zip. The file will |
| 10 be extracted in the android_tools/sdk/extras directory on the test bots. This |
| 11 script will not do anything for developers. |
| 12 |
| 13 TODO(navabi): Move this script (crbug.com/459819). |
| 14 """ |
| 15 |
| 16 import json |
| 17 import os |
| 18 import shutil |
| 19 import subprocess |
| 20 import sys |
| 21 import zipfile |
| 22 |
| 23 SCRIPT_DIR = os.path.dirname(os.path.realpath(__file__)) |
| 24 CHROME_SRC = os.path.abspath(os.path.join(SCRIPT_DIR, os.pardir)) |
| 25 sys.path.insert(0, os.path.join(SCRIPT_DIR, 'android')) |
| 26 sys.path.insert(1, os.path.join(CHROME_SRC, 'tools')) |
| 27 |
| 28 from pylib import constants |
| 29 import find_depot_tools |
| 30 |
| 31 DEPOT_PATH = find_depot_tools.add_depot_tools_to_path() |
| 32 GSUTIL_PATH = os.path.join(DEPOT_PATH, 'gsutil.py') |
| 33 SDK_EXTRAS_BUCKET = 'gs://chrome-sdk-extras' |
| 34 SDK_EXTRAS_PATH = os.path.join(constants.ANDROID_SDK_ROOT, 'extras') |
| 35 SDK_EXTRAS_JSON_FILE = os.path.join(os.path.dirname(__file__), |
| 36 'android_sdk_extras.json') |
| 37 |
| 38 |
| 39 def clean_and_extract(dir_name, package_name, zip_file): |
| 40 local_dir = '%s/%s/%s' % (SDK_EXTRAS_PATH, dir_name, package_name) |
| 41 if os.path.exists(local_dir): |
| 42 shutil.rmtree(local_dir) |
| 43 local_zip = '%s/%s' % (SDK_EXTRAS_PATH, zip_file) |
| 44 with zipfile.ZipFile(local_zip) as z: |
| 45 z.extractall(path=SDK_EXTRAS_PATH) |
| 46 |
| 47 |
| 48 def main(): |
| 49 if not os.environ.get('CHROME_HEADLESS'): |
| 50 # This is not a buildbot checkout. |
| 51 return 0 |
| 52 # Update the android_sdk_extras.json file to update downloaded packages. |
| 53 with open(SDK_EXTRAS_JSON_FILE) as json_file: |
| 54 packages = json.load(json_file) |
| 55 for package in packages: |
| 56 local_zip = '%s/%s' % (SDK_EXTRAS_PATH, package['zip']) |
| 57 if not os.path.exists(local_zip): |
| 58 package_zip = '%s/%s' % (SDK_EXTRAS_BUCKET, package['zip']) |
| 59 try: |
| 60 subprocess.check_call(['python', GSUTIL_PATH, '--force-version', '4.7', |
| 61 'cp', package_zip, local_zip]) |
| 62 except subprocess.CalledProcessError: |
| 63 print ('WARNING: Failed to download SDK packages. If this bot compiles ' |
| 64 'for Android, it may have errors.') |
| 65 return 0 |
| 66 # Always clean dir and extract zip to ensure correct contents. |
| 67 clean_and_extract(package['dir_name'], package['package'], package['zip']) |
| 68 |
| 69 |
| 70 if __name__ == '__main__': |
| 71 sys.exit(main()) |
OLD | NEW |