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 an argument that specifies the packet name in the following | |
9 format: <dir_in_sdk_extras>_<package_name>_<version>. There will be a | |
cjhopman
2014/12/30 19:32:52
this comment implies that I can have underscores i
| |
10 correpsonding bucket in google storage with that name, and it will be downloaded | |
cjhopman
2014/12/30 19:32:52
nit: correpsonding/corresponding
| |
11 to android_tools/sdk/extras/. | |
12 """ | |
13 | |
14 import os | |
15 import subprocess | |
16 import sys | |
17 | |
18 sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'android')) | |
19 from pylib import constants | |
20 | |
21 GSUTIL_PATH = os.path.join(os.path.dirname(__file__), os.pardir, os.pardir, | |
22 os.pardir, os.pardir, os.pardir, os.pardir, 'depot_tools', 'gsutil.py') | |
23 SDK_EXTRAS_BUCKET = 'gs://chrome-sdk-extras' | |
24 SDK_EXTRAS_PATH = os.path.join(constants.ANDROID_SDK_ROOT, 'extras') | |
25 | |
26 | |
27 def GetCmdOutputAndStatus(cmd_lst): | |
28 process = subprocess.Popen(cmd_lst, stdout=subprocess.PIPE) | |
29 stdout, _ = process.communicate() | |
30 return stdout, process.returncode | |
31 | |
32 def is_android_buildbot_checkout(): | |
33 if not os.path.exists(GSUTIL_PATH) or not os.path.exists(SDK_EXTRAS_PATH): | |
34 return False | |
35 stdout, rc = GetCmdOutputAndStatus([GSUTIL_PATH, 'ls', SDK_EXTRAS_BUCKET]) | |
36 # If successfully read bucket, then this must be a bot with permissions | |
37 return not rc | |
38 | |
39 def main(args): | |
40 if is_android_buildbot_checkout(): | |
41 success = True | |
42 for arg in args[1:]: | |
43 # Package is named <folder>_<package_name>_<version> | |
44 first_underscore = arg.find('_') | |
cjhopman
2014/12/30 19:32:52
should this use a different separator that wouldn'
| |
45 last_underscore = arg.rfind('_') | |
46 folder = arg[0:first_underscore] | |
47 package = arg[first_underscore+1:last_underscore] | |
48 # Package bucket is <SDK_EXTRAS_BUCKET>/<folder>_<package_name>_<version> | |
49 # and in that bucket will be the directory <folder>/<package_name> to cp. | |
50 package_bucket = '%s/%s/%s/%s' % (SDK_EXTRAS_BUCKET, arg, folder, package) | |
cjhopman
2014/12/30 19:32:52
This with the comment above is a little confusing
| |
51 package_dir = '%s/%s/%s' % (SDK_EXTRAS_PATH, folder, package) | |
52 if not os.path.exists(package_dir): | |
53 os.makedirs(package_dir) | |
54 # rsync is only supported by gsutil version 4.x | |
55 cmd_lst = [GSUTIL_PATH, '--force-version', '4.6', '-m', 'rsync', '-r', | |
56 '-d', package_bucket, package_dir] | |
57 stdout, rc = GetCmdOutputAndStatus(cmd_lst) | |
cjhopman
2014/12/30 19:32:52
how about just:
stdout = subprocess.check_output(
| |
58 success = (rc == 0) and success | |
59 if not success: | |
60 return 1 | |
61 return 0 | |
62 | |
63 | |
64 if __name__ == '__main__': | |
65 sys.exit(main(sys.argv)) | |
OLD | NEW |