| Index: build/android/adb_install_apk.py
|
| diff --git a/build/android/adb_install_apk.py b/build/android/adb_install_apk.py
|
| index 0c6fe8255ebffb0b837ea937bfc8767a1791ff32..00bd38e1821d7957efcc5ffbd0dc9bc027f6fd09 100755
|
| --- a/build/android/adb_install_apk.py
|
| +++ b/build/android/adb_install_apk.py
|
| @@ -6,87 +6,84 @@
|
|
|
| """Utility script to install APKs from the command line quickly."""
|
|
|
| -import argparse
|
| -import logging
|
| +import optparse
|
| import os
|
| import sys
|
|
|
| from pylib import constants
|
| -from pylib.device import device_blacklist
|
| from pylib.device import device_errors
|
| from pylib.device import device_utils
|
| -from pylib.utils import run_tests_helper
|
|
|
|
|
| -def main():
|
| - parser = argparse.ArgumentParser()
|
| +def AddInstallAPKOption(option_parser):
|
| + """Adds apk option used to install the APK to the OptionParser."""
|
| + option_parser.add_option('--apk',
|
| + help=('DEPRECATED The name of the apk containing the'
|
| + ' application (with the .apk extension).'))
|
| + option_parser.add_option('--apk_package',
|
| + help=('DEPRECATED The package name used by the apk '
|
| + 'containing the application.'))
|
| + option_parser.add_option('--keep_data',
|
| + action='store_true',
|
| + default=False,
|
| + help=('Keep the package data when installing '
|
| + 'the application.'))
|
| + option_parser.add_option('--debug', action='store_const', const='Debug',
|
| + dest='build_type',
|
| + default=os.environ.get('BUILDTYPE', 'Debug'),
|
| + help='If set, run test suites under out/Debug. '
|
| + 'Default is env var BUILDTYPE or Debug')
|
| + option_parser.add_option('--release', action='store_const', const='Release',
|
| + dest='build_type',
|
| + help='If set, run test suites under out/Release. '
|
| + 'Default is env var BUILDTYPE or Debug.')
|
| + option_parser.add_option('-d', '--device', dest='device',
|
| + help='Target device for apk to install on.')
|
|
|
| - apk_group = parser.add_mutually_exclusive_group(required=True)
|
| - apk_group.add_argument('--apk', dest='apk_name',
|
| - help='DEPRECATED The name of the apk containing the'
|
| - ' application (with the .apk extension).')
|
| - apk_group.add_argument('apk_path', nargs='?',
|
| - help='The path to the APK to install.')
|
|
|
| - # TODO(jbudorick): Remove once no clients pass --apk_package
|
| - parser.add_argument('--apk_package', help='DEPRECATED unused')
|
| - parser.add_argument('--keep_data',
|
| - action='store_true',
|
| - default=False,
|
| - help='Keep the package data when installing '
|
| - 'the application.')
|
| - parser.add_argument('--debug', action='store_const', const='Debug',
|
| - dest='build_type',
|
| - default=os.environ.get('BUILDTYPE', 'Debug'),
|
| - help='If set, run test suites under out/Debug. '
|
| - 'Default is env var BUILDTYPE or Debug')
|
| - parser.add_argument('--release', action='store_const', const='Release',
|
| - dest='build_type',
|
| - help='If set, run test suites under out/Release. '
|
| - 'Default is env var BUILDTYPE or Debug.')
|
| - parser.add_argument('-d', '--device', dest='device',
|
| - help='Target device for apk to install on.')
|
| - parser.add_argument('-v', '--verbose', action='count',
|
| - help='Enable verbose logging.')
|
| +def ValidateInstallAPKOption(option_parser, options, args):
|
| + """Validates the apk option and potentially qualifies the path."""
|
| + if not options.apk:
|
| + if len(args) > 1:
|
| + options.apk = args[1]
|
| + else:
|
| + option_parser.error('apk target not specified.')
|
|
|
| - args = parser.parse_args()
|
| + if not options.apk.endswith('.apk'):
|
| + options.apk += '.apk'
|
|
|
| - run_tests_helper.SetLogLevel(args.verbose)
|
| - constants.SetBuildType(args.build_type)
|
| + if not os.path.exists(options.apk):
|
| + options.apk = os.path.join(constants.GetOutDirectory(), 'apks',
|
| + options.apk)
|
|
|
| - if args.apk_path:
|
| - apk = args.apk_path
|
| - else:
|
| - apk = args.apk_name
|
| - if not apk.endswith('.apk'):
|
| - apk += '.apk'
|
| - if not os.path.exists(apk):
|
| - apk = os.path.join(constants.GetOutDirectory(), 'apks', apk)
|
| +
|
| +def main(argv):
|
| + parser = optparse.OptionParser()
|
| + parser.set_usage("usage: %prog [options] target")
|
| + AddInstallAPKOption(parser)
|
| + options, args = parser.parse_args(argv)
|
| +
|
| + if len(args) > 1 and options.apk:
|
| + parser.error("Appending the apk as argument can't be used with --apk.")
|
| + elif len(args) > 2:
|
| + parser.error("Too many arguments.")
|
| +
|
| + constants.SetBuildType(options.build_type)
|
| + ValidateInstallAPKOption(parser, options, args)
|
|
|
| devices = device_utils.DeviceUtils.HealthyDevices()
|
|
|
| - if args.device:
|
| - devices = [d for d in devices if d == args.device]
|
| + if options.device:
|
| + devices = [d for d in devices if d == options.device]
|
| if not devices:
|
| - raise device_errors.DeviceUnreachableError(args.device)
|
| + raise device_errors.DeviceUnreachableError(options.device)
|
| elif not devices:
|
| raise device_errors.NoDevicesError()
|
|
|
| - def blacklisting_install(device):
|
| - try:
|
| - device.Install(apk, reinstall=args.keep_data)
|
| - except device_errors.CommandFailedError:
|
| - logging.exception('Failed to install %s', args.apk)
|
| - device_blacklist.ExtendBlacklist([str(device)])
|
| - logging.warning('Blacklisting %s', str(device))
|
| - except device_errors.CommandTimeoutError:
|
| - logging.exception('Timed out while installing %s', args.apk)
|
| - device_blacklist.ExtendBlacklist([str(device)])
|
| - logging.warning('Blacklisting %s', str(device))
|
| -
|
| - device_utils.DeviceUtils.parallel(devices).pMap(blacklisting_install)
|
| + device_utils.DeviceUtils.parallel(devices).Install(
|
| + options.apk, reinstall=options.keep_data)
|
|
|
|
|
| if __name__ == '__main__':
|
| - sys.exit(main())
|
| + sys.exit(main(sys.argv))
|
|
|
|
|