OLD | NEW |
| (Empty) |
1 #!/usr/bin/env python | |
2 # | |
3 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
4 # Use of this source code is governed by a BSD-style license that can be | |
5 # found in the LICENSE file. | |
6 | |
7 """Utility script to install APKs from the command line quickly.""" | |
8 | |
9 import argparse | |
10 import glob | |
11 import logging | |
12 import os | |
13 import sys | |
14 | |
15 from pylib import constants | |
16 from pylib.device import device_blacklist | |
17 from pylib.device import device_errors | |
18 from pylib.device import device_utils | |
19 from pylib.utils import apk_helper | |
20 from pylib.utils import run_tests_helper | |
21 | |
22 | |
23 def main(): | |
24 parser = argparse.ArgumentParser() | |
25 | |
26 apk_group = parser.add_mutually_exclusive_group(required=True) | |
27 apk_group.add_argument('--apk', dest='apk_name', | |
28 help='DEPRECATED The name of the apk containing the' | |
29 ' application (with the .apk extension).') | |
30 apk_group.add_argument('apk_path', nargs='?', | |
31 help='The path to the APK to install.') | |
32 | |
33 # TODO(jbudorick): Remove once no clients pass --apk_package | |
34 parser.add_argument('--apk_package', help='DEPRECATED unused') | |
35 parser.add_argument('--split', | |
36 action='append', | |
37 dest='splits', | |
38 help='A glob matching the apk splits. ' | |
39 'Can be specified multiple times.') | |
40 parser.add_argument('--keep_data', | |
41 action='store_true', | |
42 default=False, | |
43 help='Keep the package data when installing ' | |
44 'the application.') | |
45 parser.add_argument('--debug', action='store_const', const='Debug', | |
46 dest='build_type', | |
47 default=os.environ.get('BUILDTYPE', 'Debug'), | |
48 help='If set, run test suites under out/Debug. ' | |
49 'Default is env var BUILDTYPE or Debug') | |
50 parser.add_argument('--release', action='store_const', const='Release', | |
51 dest='build_type', | |
52 help='If set, run test suites under out/Release. ' | |
53 'Default is env var BUILDTYPE or Debug.') | |
54 parser.add_argument('-d', '--device', dest='device', | |
55 help='Target device for apk to install on.') | |
56 parser.add_argument('-v', '--verbose', action='count', | |
57 help='Enable verbose logging.') | |
58 | |
59 args = parser.parse_args() | |
60 | |
61 run_tests_helper.SetLogLevel(args.verbose) | |
62 constants.SetBuildType(args.build_type) | |
63 | |
64 apk = args.apk_path or args.apk_name | |
65 if not apk.endswith('.apk'): | |
66 apk += '.apk' | |
67 if not os.path.exists(apk): | |
68 apk = os.path.join(constants.GetOutDirectory(), 'apks', apk) | |
69 if not os.path.exists(apk): | |
70 parser.error('%s not found.' % apk) | |
71 | |
72 if args.splits: | |
73 splits = [] | |
74 base_apk_package = apk_helper.ApkHelper(apk).GetPackageName() | |
75 for split_glob in args.splits: | |
76 apks = [f for f in glob.glob(split_glob) if f.endswith('.apk')] | |
77 if not apks: | |
78 logging.warning('No apks matched for %s.' % split_glob) | |
79 for f in apks: | |
80 helper = apk_helper.ApkHelper(f) | |
81 if (helper.GetPackageName() == base_apk_package | |
82 and helper.GetSplitName()): | |
83 splits.append(f) | |
84 | |
85 devices = device_utils.DeviceUtils.HealthyDevices() | |
86 | |
87 if args.device: | |
88 devices = [d for d in devices if d == args.device] | |
89 if not devices: | |
90 raise device_errors.DeviceUnreachableError(args.device) | |
91 elif not devices: | |
92 raise device_errors.NoDevicesError() | |
93 | |
94 def blacklisting_install(device): | |
95 try: | |
96 if args.splits: | |
97 device.InstallSplitApk(apk, splits, reinstall=args.keep_data) | |
98 else: | |
99 device.Install(apk, reinstall=args.keep_data) | |
100 except device_errors.CommandFailedError: | |
101 logging.exception('Failed to install %s', args.apk_name) | |
102 device_blacklist.ExtendBlacklist([str(device)]) | |
103 logging.warning('Blacklisting %s', str(device)) | |
104 except device_errors.CommandTimeoutError: | |
105 logging.exception('Timed out while installing %s', args.apk_name) | |
106 device_blacklist.ExtendBlacklist([str(device)]) | |
107 logging.warning('Blacklisting %s', str(device)) | |
108 | |
109 device_utils.DeviceUtils.parallel(devices).pMap(blacklisting_install) | |
110 | |
111 | |
112 if __name__ == '__main__': | |
113 sys.exit(main()) | |
114 | |
OLD | NEW |