build/download_sdk_extras.py

   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())