Index: build/android/screenshot.py |
diff --git a/build/android/screenshot.py b/build/android/screenshot.py |
index 4c1e247fcef09423c4b183458e005807b07482de..c48a2553fbd0f6ffe967d61bb611b1b1c3559f06 100755 |
--- a/build/android/screenshot.py |
+++ b/build/android/screenshot.py |
@@ -12,8 +12,7 @@ import os |
import sys |
from pylib import screenshot |
-from pylib.device import adb_wrapper |
-from pylib.device import device_filter |
+from pylib.device import device_errors |
from pylib.device import device_utils |
def _PrintMessage(heading, eol='\n'): |
@@ -67,22 +66,25 @@ def main(): |
(options, args) = parser.parse_args() |
+ if len(args) > 1: |
+ parser.error('Too many positional arguments.') |
+ host_file = args[0] if args else options.file |
+ |
if options.verbose: |
logging.getLogger().setLevel(logging.DEBUG) |
- devices = adb_wrapper.AdbWrapper.Devices( |
- filters=device_filter.DefaultFilters()) |
- |
- if not options.device and len(devices) > 1: |
- parser.error('Multiple devices are attached. ' |
- 'Please specify device serial number with --device.') |
- elif not options.device and len(devices) == 1: |
- options.device = devices[0] |
+ devices = device_utils.DeviceUtils.HealthyDevices() |
- if len(args) > 1: |
- parser.error('Too many positional arguments.') |
- host_file = args[0] if args else options.file |
- device = device_utils.DeviceUtils(options.device) |
+ if not options.device: |
+ if len(devices) > 1: |
+ parser.error('Multiple devices are attached. ' |
+ 'Please specify device serial number with --device.') |
+ elif len(devices) == 1: |
+ device = devices[0] |
+ else: |
+ raise device_errors.NoDevicesError() |
+ else: |
+ device = device_utils.DeviceUtils(options.device) |
if options.video: |
_CaptureVideo(device, host_file, options) |