[build/android] Update adb_command_line.py to use flag_changer

build/android/adb_command_line.py

 #!/usr/bin/python
 # Copyright 2015 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """Utility for reading / writing command-line flag files on device(s)."""
 
 import argparse
 import os
+import posixpath
 import sys
 
 import devil_chromium
 
 from devil.android import device_utils
-from devil.android import device_errors
+from devil.android import flag_changer
 from devil.utils import cmd_helper
 
 
 def main():
   parser = argparse.ArgumentParser(description=__doc__)
-  parser.usage = '''%(prog)s --device-path PATH [--device SERIAL] [flags...]
+  parser.usage = '''%(prog)s [--filename FILENAME] [--device SERIAL] [flags...]
 
 No flags: Prints existing command-line file.
 Empty string: Deletes command-line file.
 Otherwise: Writes command-line file.
 
 '''
   parser.add_argument('-d', '--device', dest='devices', action='append',
                       default=[], help='Target device serial (repeatable).')
-  parser.add_argument('--device-path', required=True,
-                      help='Remote path to flags file.')
+  parser.add_argument('--filename', default='chrome-command-line',

[jbudorick, 2017/01/13 20:50:00]

Why'd you make this a default as opposed to having it encoded in the wrapped
scripts?

[perezju, 2017/01/16 14:48:20]

Ok. Changed. Also found more wrapping scripts that needed updating.

Btw, is this python script used in other places (e.g. recipes?) that should also
be changed?

+                      help='Name of file where to store flags on the device'
+                      ' (default: %(default)s).')
+  parser.add_argument('--device-path', help='(deprecated) No longer needed to'
+                      ' supply a device path.')
   parser.add_argument('-e', '--executable', dest='executable', default='chrome',
-                      help='Name of the executable.')
+                      help='(deprecated) No longer used.')
   parser.add_argument('--adb-path', type=os.path.abspath,
                       help='Path to the adb binary.')
   args, remote_args = parser.parse_known_args()
 
+  if args.device_path:
+    args.filename = posixpath.basename(args.device_path)
+    print ('warning: --device-path option is deprecated,'
+           ' --filename %s is now enough.'
+           % cmd_helper.SingleQuote(args.filename))
+
   devil_chromium.Initialize(adb_path=args.adb_path)
 
-  as_root = not args.device_path.startswith('/data/local/tmp/')
-
   devices = device_utils.DeviceUtils.HealthyDevices(device_arg=args.devices,
                                                     default_retries=0)
   all_devices = device_utils.DeviceUtils.parallel(devices)
 
-  def print_args():
-    def read_flags(device):
-      try:
-        return device.ReadFile(args.device_path, as_root=as_root).rstrip()
-      except device_errors.CommandFailedError:
-        return ''  # File might not exist.
+  if not remote_args:
+    # No args == do not update, just print flags.
+    remote_args = None
+    action = ''
+  elif len(remote_args) == 1 and not remote_args[0]:
+    # Single empty string arg == delete flags
+    remote_args = []
+    action = 'Deleted flags. '
+  else:
+    action = 'Updated flags. '
 
-    descriptions = all_devices.pMap(lambda d: d.build_description).pGet(None)
-    flags = all_devices.pMap(read_flags).pGet(None)
-    for d, desc, flags in zip(devices, descriptions, flags):
-      print '  %s (%s): %r' % (d, desc, flags)
+  def update_flags(device):
+    changer = flag_changer.FlagChanger(device, args.filename)
+    if remote_args is not None:
+      changer.ReplaceFlags(remote_args)
+    return (device, device.build_description, changer.GetCurrentFlags())
 
-  # No args == print flags.
-  if not remote_args:
-    print 'Existing flags (in %s):' % args.device_path
-    print_args()
-    return 0
+  updated_values = all_devices.pMap(update_flags).pGet(None)
 
-  # Empty string arg == delete flags file.
-  if len(remote_args) == 1 and not remote_args[0]:
-    def delete_flags(device):
-      device.RunShellCommand(['rm', '-f', args.device_path], as_root=as_root)
-    all_devices.pMap(delete_flags).pGet(None)
-    print 'Deleted %s' % args.device_path
-    return 0
-
-  # Set flags.
-  quoted_args = ' '.join(cmd_helper.SingleQuote(x) for x in remote_args)
-  flags_str = ' '.join([args.executable, quoted_args])
-
-  def write_flags(device):
-    device.WriteFile(args.device_path, flags_str, as_root=as_root)
-    device.RunShellCommand(['chmod', '0664', args.device_path], as_root=as_root)
-
-  all_devices.pMap(write_flags).pGet(None)
-  print 'Wrote flags to %s' % args.device_path
-  print_args()
-  return 0

[jbudorick, 2017/01/13 20:50:00]

nit: this should still return 0

[perezju, 2017/01/16 14:48:20]

Done.

+  print '%sCurrent flags (in %s):' % (action, args.filename)
+  for d, desc, flags in updated_values:

[jbudorick, 2017/01/13 20:49:59]

Nice!

+    if flags:
+      # Shell-quote flags for easy copy/paste as new args on the terminal.
+      quoted_flags = ' '.join(cmd_helper.SingleQuote(f) for f in flags)
+    else:
+      quoted_flags = '( empty )'
+    print '  %s (%s): %r' % (d, desc, quoted_flags)
 
 
 if __name__ == '__main__':
   sys.exit(main())