| Index: build/android/enable_asserts.py
|
| diff --git a/build/android/enable_asserts.py b/build/android/enable_asserts.py
|
| index 8fb7dca4725fe884af73115b0751d68d74f99940..0e30bc2a1736d529511b038007f6e9bc707c2b7e 100755
|
| --- a/build/android/enable_asserts.py
|
| +++ b/build/android/enable_asserts.py
|
| @@ -6,37 +6,33 @@
|
|
|
| """Enables dalvik vm asserts in the android device."""
|
|
|
| -import argparse
|
| +from pylib import android_commands
|
| +from pylib.device import device_utils
|
| +import optparse
|
| import sys
|
|
|
| -from pylib.device import device_utils
|
|
|
| -
|
| -def main():
|
| - parser = argparse.ArgumentParser()
|
| -
|
| - set_asserts_group = parser.add_mutually_exclusive_group(required=True)
|
| - set_asserts_group.add_argument(
|
| - '--enable_asserts', dest='set_asserts', action='store_true',
|
| +def main(argv):
|
| + option_parser = optparse.OptionParser()
|
| + option_parser.add_option('--enable_asserts', dest='set_asserts',
|
| + action='store_true', default=None,
|
| help='Sets the dalvik.vm.enableassertions property to "all"')
|
| - set_asserts_group.add_argument(
|
| - '--disable_asserts', dest='set_asserts', action='store_false',
|
| + option_parser.add_option('--disable_asserts', dest='set_asserts',
|
| + action='store_false', default=None,
|
| help='Removes the dalvik.vm.enableassertions property')
|
| -
|
| - args = parser.parse_args()
|
| + options, _ = option_parser.parse_args(argv)
|
|
|
| # TODO(jbudorick): Accept optional serial number and run only for the
|
| # specified device when present.
|
| - devices = device_utils.DeviceUtils.parallel()
|
| -
|
| - def set_java_asserts_and_restart(device):
|
| - if device.SetJavaAsserts(args.set_asserts):
|
| - device.RunShellCommand('stop')
|
| - device.RunShellCommand('start')
|
| -
|
| - devices.pMap(set_java_asserts_and_restart)
|
| - return 0
|
| + devices = android_commands.GetAttachedDevices()
|
| + for device in [device_utils.DeviceUtils(serial) for serial in devices]:
|
| + if options.set_asserts != None:
|
| + if device.SetJavaAsserts(options.set_asserts):
|
| + # TODO(jbudorick) How to best do shell restarts after the
|
| + # android_commands refactor?
|
| + device.RunShellCommand('stop')
|
| + device.RunShellCommand('start')
|
|
|
|
|
| if __name__ == '__main__':
|
| - sys.exit(main())
|
| + main(sys.argv)
|
|
|