OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright 2015 The Chromium Authors. All rights reserved. | 2 # Copyright 2015 The Chromium Authors. All rights reserved. |
3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
5 | 5 |
6 '''Checks the status of an Android SDK package. | 6 '''Checks the status of an Android SDK package. |
7 | 7 |
8 Verifies the given package has been installed from the Android SDK Manager and | 8 TODO(dgn) replaced by a direct update mechanism: http://crbug.com/541727 |
9 that its version is at least the minimum version required by the project | 9 This file is now a placeholder until removal after 2 sided patches. |
10 configuration. | |
11 ''' | 10 ''' |
12 | |
13 import argparse | |
14 import json | |
15 import os | |
16 import re | |
17 import sys | |
18 | |
19 | |
20 COLORAMA_ROOT = os.path.join(os.path.dirname(__file__), | |
21 os.pardir, 'third_party', 'colorama', 'src') | |
22 | |
23 sys.path.append(COLORAMA_ROOT) | |
24 import colorama | |
25 | |
26 | |
27 UDPATE_SCRIPT_PATH = 'build/install-android-sdks.sh' | |
28 | |
29 SDK_EXTRAS_JSON_FILE = os.path.join(os.path.dirname(__file__), | |
30 'android_sdk_extras.json') | |
31 | |
32 PACKAGE_VERSION_PATTERN = r'^Pkg\.Revision=(?P<version>\d+).*$' | |
33 | |
34 PKG_NOT_FOUND_MSG = ('Error while checking Android SDK extras versions. ' | |
35 'Could not find the "{package_id}" package in ' | |
36 '{checked_location}. Please run {script} to download it.') | |
37 UPDATE_NEEDED_MSG = ('Error while checking Android SDK extras versions. ' | |
38 'Version {minimum_version} or greater is required for the ' | |
39 'package "{package_id}". Version {actual_version} found. ' | |
40 'Please run {script} to update it.') | |
41 REQUIRED_VERSION_ERROR_MSG = ('Error while checking Android SDK extras ' | |
42 'versions. ' | |
43 'Could not retrieve the required version for ' | |
44 'package "{package_id}".') | |
45 | |
46 | |
47 def main(): | |
48 parser = argparse.ArgumentParser(description=__doc__) | |
49 parser.add_argument('--package-id', | |
50 help=('id of the package to check for. The list of ' | |
51 'available packages and their ids can be obtained ' | |
52 'by running ' | |
53 'third_party/android_tools/sdk/tools/android list ' | |
54 'sdk --extended')) | |
55 parser.add_argument('--package-location', | |
56 help='path to the package\'s expected install location.', | |
57 metavar='DIR') | |
58 parser.add_argument('--stamp', | |
59 help=('if specified, a stamp file will be created at the ' | |
60 'provided location.'), | |
61 metavar='FILE') | |
62 | |
63 args = parser.parse_args() | |
64 | |
65 if not ShouldSkipVersionCheck(): | |
66 minimum_version = GetRequiredMinimumVersion(args.package_id) | |
67 CheckPackageVersion(args.package_id, args.package_location, minimum_version) | |
68 | |
69 # Create the stamp file. | |
70 if args.stamp: | |
71 with open(args.stamp, 'a'): | |
72 os.utime(args.stamp, None) | |
73 | |
74 sys.exit(0) | |
75 | |
76 def ExitError(msg): | |
77 sys.exit(colorama.Fore.MAGENTA + colorama.Style.BRIGHT + msg + | |
78 colorama.Style.RESET_ALL) | |
79 | |
80 | |
81 def GetRequiredMinimumVersion(package_id): | |
82 with open(SDK_EXTRAS_JSON_FILE, 'r') as json_file: | |
83 packages = json.load(json_file) | |
84 | |
85 for package in packages: | |
86 if package['package_id'] == package_id: | |
87 return int(package['version'].split('.')[0]) | |
88 | |
89 ExitError(REQUIRED_VERSION_ERROR_MSG.format(package_id=package_id)) | |
90 | |
91 | |
92 def CheckPackageVersion(pkg_id, location, minimum_version): | |
93 version_file_path = os.path.join(location, 'source.properties') | |
94 # Extracts the version of the package described by the property file. We only | |
95 # care about the major version number here. | |
96 version_pattern = re.compile(PACKAGE_VERSION_PATTERN, re.MULTILINE) | |
97 | |
98 if not os.path.isfile(version_file_path): | |
99 ExitError(PKG_NOT_FOUND_MSG.format( | |
100 package_id=pkg_id, | |
101 checked_location=location, | |
102 script=UDPATE_SCRIPT_PATH)) | |
103 | |
104 with open(version_file_path, 'r') as f: | |
105 match = version_pattern.search(f.read()) | |
106 | |
107 if not match: | |
108 ExitError(PKG_NOT_FOUND_MSG.format( | |
109 package_id=pkg_id, | |
110 checked_location=location, | |
111 script=UDPATE_SCRIPT_PATH)) | |
112 | |
113 pkg_version = int(match.group('version')) | |
114 if pkg_version < minimum_version: | |
115 ExitError(UPDATE_NEEDED_MSG.format( | |
116 package_id=pkg_id, | |
117 minimum_version=minimum_version, | |
118 actual_version=pkg_version, | |
119 script=UDPATE_SCRIPT_PATH)) | |
120 | |
121 # Everything looks ok, print nothing. | |
122 | |
123 def ShouldSkipVersionCheck(): | |
124 ''' | |
125 Bots should not run the version check, since they download the sdk extras | |
126 in a different way. | |
127 ''' | |
128 return bool(os.environ.get('CHROME_HEADLESS')) | |
129 | |
130 if __name__ == '__main__': | |
131 main() | |
OLD | NEW |