add GetBattorPathFromPhoneSerial to find_usb_devices

devil/devil/utils/update_mapping.py

Index: devil/devil/utils/update_mapping.py
diff --git a/devil/devil/utils/update_mapping.py b/devil/devil/utils/update_mapping.py
new file mode 100755
index 0000000000000000000000000000000000000000..95aba84b850e5c97c28a575d4987cd087ca5926f
--- /dev/null
+++ b/devil/devil/utils/update_mapping.py
@@ -0,0 +1,45 @@
+#!/usr/bin/env python
+
+# Copyright 2016 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.
+
+import argparse
+import sys
+
+from devil.utils import battor_device_mapping
+
+def parse_options():
+  """Parses and checks the command-line options.
+
+  Returns:
+    A tuple containing the options structure.
+  """
+  usage = 'Usage: ./update_mapping.py [options]'
+  desc = ('Example: ./update_mapping.py -o mapping.json.\n'
+  'This script generates and stores a file that gives the\n'
+  'mapping between phone serial numbers and BattOr serial numbers\n'
+  'Mapping is based on which physical ports on the USB hubs the\n'
+  'devices are plugged in to. For instance, if there are two hubs,\n'
+  'the phone connected to port N on the first hub is mapped to the\n'
+  'BattOr connected to port N on the second hub, for each N.')
+  parser = argparse.ArgumentParser(usage=usage, description=desc)
+  parser.add_argument('-o', '--output', dest='out_file',
+                      default='mapping.json', type=str,
+                      action='store', help='mapping file name')
+  parser.add_argument('-u', '--hubs', dest='hub_types',
+                      default='plugable_7port', type=str,
+                      action='store', help='List of usb hub types. '
+                      'Supported types:'

[jbudorick, 2016/03/31 20:57:31]

Would choices be helpful here?
https://docs.python.org/3/library/argparse.html#choices

[alexandermont, 2016/03/31 22:11:18]

Done

+                      ' plugable_7port = Plugable 7-Port Hub')
+  options = parser.parse_args()
+  options.hub_types = options.hub_types.split(',')
+  return options
+
+def main():
+  options = parse_options()
+  battor_device_mapping.GenerateSerialMapFile(options.out_file,
+                                              options.hub_types)
+
+if __name__ == "__main__":
+  sys.exit(main())