| Index: build/android/provision_devices.py
|
| diff --git a/build/android/provision_devices.py b/build/android/provision_devices.py
|
| index 5f00b31e8a817440b99d845b6745401c6d46c947..f70526f7bc3cb526358931067bba0b4a2ca5f78d 100755
|
| --- a/build/android/provision_devices.py
|
| +++ b/build/android/provision_devices.py
|
| @@ -22,6 +22,7 @@ import time
|
| from pylib import android_commands
|
| from pylib import constants
|
| from pylib import device_settings
|
| +from pylib.device import battery_utils
|
| from pylib.device import device_blacklist
|
| from pylib.device import device_errors
|
| from pylib.device import device_utils
|
| @@ -165,7 +166,8 @@ def WipeDeviceIfPossible(device, timeout, options):
|
|
|
| def ChargeDeviceToLevel(device, level):
|
| def device_charged():
|
| - battery_level = device.GetBatteryInfo().get('level')
|
| + battery = battery_utils.BatteryUtils(device)
|
| + battery_level = battery.GetBatteryInfo().get('level')
|
| if battery_level is None:
|
| logging.warning('Unable to find current battery level.')
|
| battery_level = 100
|
| @@ -208,7 +210,8 @@ def ProvisionDevice(device, options):
|
| device, device_settings.NETWORK_DISABLED_SETTINGS)
|
| if options.min_battery_level is not None:
|
| try:
|
| - device.SetCharging(True)
|
| + battery = battery_utils.BatteryUtils(device)
|
| + battery.SetCharging(True)
|
| ChargeDeviceToLevel(device, options.min_battery_level)
|
| except device_errors.CommandFailedError as e:
|
| logging.exception('Unable to charge device to specified level.')
|
|
|