Wait for perfbot devices to charge between runs.

build/android/buildbot/bb_device_status_check.py

 #!/usr/bin/env python
 #
 # Copyright 2013 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.
 
 """A class to keep track of devices across builds and report state."""
 import logging
 import optparse
 import os
@@ skipping 34 matching lines @@
     boolean indicating whether or not device can be used for testing.
   """
 
   device_adb = device_utils.DeviceUtils(serial)
   device_type = device_adb.old_interface.GetBuildProduct()
   device_build = device_adb.old_interface.GetBuildId()
   device_build_type = device_adb.old_interface.GetBuildType()
   device_product_name = device_adb.old_interface.GetProductName()
 
   try:
-    battery = device_adb.old_interface.GetBatteryInfo()
+    battery_info = device_adb.old_interface.GetBatteryInfo()
   except Exception as e:
-    battery = None
+    battery_info = {}
     logging.error('Unable to obtain battery info for %s, %s', serial, e)
 
   def _GetData(re_expression, line, lambda_function=lambda x:x):
     if not line:
       return 'Unknown'
     found = re.findall(re_expression, line)
     if found and len(found):
       return lambda_function(found[0])
     return 'Unknown'
 
-  def _GetBatteryInfo(battery):
-    if not battery:
-      return 'No battery info.'
-    battery_info = battery.split('\n')
-    return battery_info[0] + '\n  '.join(battery_info[1:])
-
-  ac_power = _GetData('AC powered: (\w+)', battery)
-  battery_level = _GetData('level: (\d+)', battery)
+  battery_level = int(battery_info.get('level', 100))
   imei_slice = _GetData('Device ID = (\d+)',
                         device_adb.old_interface.GetSubscriberInfo(),
                         lambda x: x[-6:])
   report = ['Device %s (%s)' % (serial, device_type),
             '  Build: %s (%s)' %
               (device_build, device_adb.old_interface.GetBuildFingerprint()),
-            '  %s' % _GetBatteryInfo(battery),
+            '  Current Battery Service state: ',
+            '\n'.join(['    %s: %s' % (k, v)
+                       for k, v in battery_info.iteritems()]),
             '  IMEI slice: %s' % imei_slice,
             '  Wifi IP: %s' % device_adb.old_interface.GetWifiIP(),
             '']
 
   errors = []
   if battery_level < 15:
     errors += ['Device critically low in battery. Turning off device.']
   if not options.no_provisioning_check:
     setup_wizard_disabled = (
         device_adb.old_interface.GetSetupWizardStatus() == 'DISABLED')
     if not setup_wizard_disabled and device_build_type != 'user':
       errors += ['Setup wizard not disabled. Was it provisioned correctly?']
-  if device_product_name == 'mantaray' and ac_power != 'true':
+  if (device_product_name == 'mantaray' and
+      battery_info.get('AC powered', None) != 'true'):
     errors += ['Mantaray device not connected to AC power.']
 
   # Turn off devices with low battery.
   if battery_level < 15:
     try:
       device_adb.EnableRoot()
     except device_errors.CommandFailedError as e:
       # Attempt shutdown anyway.
       # TODO(jbudorick) Handle this exception appropriately after interface
       #                 conversions are finished.
@@ skipping 236 matching lines @@
     # devices with critically low battery. Remove those devices from testing,
     # allowing build to continue with good devices.
     return 2
 
   if not devices:
     return 1
 
 
 if __name__ == '__main__':
   sys.exit(main())