OLD | NEW |
---|---|
1 # Copyright 2013 The Chromium Authors. All rights reserved. | 1 # Copyright 2013 The Chromium Authors. All rights reserved. |
2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 """This module wraps Android's adb tool. | 5 """This module wraps Android's adb tool. |
6 | 6 |
7 This is a thin wrapper around the adb interface. Any additional complexity | 7 This is a thin wrapper around the adb interface. Any additional complexity |
8 should be delegated to a higher level (ex. DeviceUtils). | 8 should be delegated to a higher level (ex. DeviceUtils). |
9 """ | 9 """ |
10 | 10 |
11 import collections | 11 import collections |
12 import errno | 12 import errno |
13 import logging | 13 import logging |
14 import os | 14 import os |
15 | 15 |
16 from pylib import cmd_helper | 16 from pylib import cmd_helper |
17 from pylib import constants | 17 from pylib import constants |
18 from pylib.device import decorators | 18 from pylib.device import decorators |
19 from pylib.device import device_errors | 19 from pylib.device import device_errors |
20 from pylib.device import device_filter | |
20 from pylib.utils import timeout_retry | 21 from pylib.utils import timeout_retry |
21 | 22 |
22 | 23 |
23 _DEFAULT_TIMEOUT = 30 | 24 _DEFAULT_TIMEOUT = 30 |
24 _DEFAULT_RETRIES = 2 | 25 _DEFAULT_RETRIES = 2 |
25 | 26 |
26 | 27 |
27 def _VerifyLocalFileExists(path): | 28 def _VerifyLocalFileExists(path): |
28 """Verifies a local file exists. | 29 """Verifies a local file exists. |
29 | 30 |
(...skipping 121 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
151 @classmethod | 152 @classmethod |
152 def KillServer(cls, timeout=_DEFAULT_TIMEOUT, retries=_DEFAULT_RETRIES): | 153 def KillServer(cls, timeout=_DEFAULT_TIMEOUT, retries=_DEFAULT_RETRIES): |
153 cls._RunAdbCmd(['kill-server'], timeout=timeout, retries=retries) | 154 cls._RunAdbCmd(['kill-server'], timeout=timeout, retries=retries) |
154 | 155 |
155 @classmethod | 156 @classmethod |
156 def StartServer(cls, timeout=_DEFAULT_TIMEOUT, retries=_DEFAULT_RETRIES): | 157 def StartServer(cls, timeout=_DEFAULT_TIMEOUT, retries=_DEFAULT_RETRIES): |
157 # CPU affinity is used to reduce adb instability http://crbug.com/268450 | 158 # CPU affinity is used to reduce adb instability http://crbug.com/268450 |
158 cls._RunAdbCmd(['start-server'], timeout=timeout, retries=retries, | 159 cls._RunAdbCmd(['start-server'], timeout=timeout, retries=retries, |
159 cpu_affinity=0) | 160 cpu_affinity=0) |
160 | 161 |
161 # TODO(craigdh): Determine the filter criteria that should be supported. | |
162 @classmethod | 162 @classmethod |
163 def GetDevices(cls, timeout=_DEFAULT_TIMEOUT, retries=_DEFAULT_RETRIES): | 163 def GetDevices(cls, filters=None, timeout=_DEFAULT_TIMEOUT, |
164 retries=_DEFAULT_RETRIES): | |
165 """DEPRECATED. Refer to Devices(...) below.""" | |
166 # TODO(jbudorick): Remove this function once no more clients are using it. | |
167 return cls.Devices(filters=filters, timeout=timeout, retries=retries) | |
168 | |
169 @classmethod | |
170 def Devices(cls, filters=None, timeout=_DEFAULT_TIMEOUT, | |
171 retries=_DEFAULT_RETRIES): | |
164 """Get the list of active attached devices. | 172 """Get the list of active attached devices. |
165 | 173 |
166 Args: | 174 Args: |
175 filters: (optional) A list of binary functions that take an AdbWrapper | |
176 instance and a string description. Any device for which all provided | |
177 filter functions do not return True will not be included in the | |
178 returned list. | |
167 timeout: (optional) Timeout per try in seconds. | 179 timeout: (optional) Timeout per try in seconds. |
168 retries: (optional) Number of retries to attempt. | 180 retries: (optional) Number of retries to attempt. |
169 | 181 |
170 Yields: | 182 Yields: |
171 AdbWrapper instances. | 183 AdbWrapper instances. |
172 """ | 184 """ |
173 output = cls._RunAdbCmd(['devices'], timeout=timeout, retries=retries) | 185 output = cls._RunAdbCmd(['devices'], timeout=timeout, retries=retries) |
174 lines = [line.split() for line in output.splitlines()] | 186 lines = (line.split() for line in output.splitlines()) |
175 return [AdbWrapper(line[0]) for line in lines | 187 devices = (AdbWrapper(line[0]) for line in lines if len(line) == 2) |
176 if len(line) == 2 and line[1] == 'device'] | 188 |
189 def matches_all_filters(device): | |
190 for f in filters or (): | |
191 if not f(device): | |
192 logging.info('Device %s failed filter %s', device.GetDeviceSerial(), | |
193 f.__name__) | |
194 return False | |
195 return True | |
196 | |
197 return [d for d in devices if matches_all_filters(d)] | |
perezju
2015/04/16 08:57:55
optional nit: I would go for something a bit more
jbudorick
2015/04/16 13:53:14
landing as-is
| |
177 | 198 |
178 def GetDeviceSerial(self): | 199 def GetDeviceSerial(self): |
179 """Gets the device serial number associated with this object. | 200 """Gets the device serial number associated with this object. |
180 | 201 |
181 Returns: | 202 Returns: |
182 Device serial number as a string. | 203 Device serial number as a string. |
183 """ | 204 """ |
184 return self._device_serial | 205 return self._device_serial |
185 | 206 |
186 def Push(self, local, remote, timeout=60*5, retries=_DEFAULT_RETRIES): | 207 def Push(self, local, remote, timeout=60*5, retries=_DEFAULT_RETRIES): |
(...skipping 336 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
523 """Restarts the adbd daemon with root permissions, if possible. | 544 """Restarts the adbd daemon with root permissions, if possible. |
524 | 545 |
525 Args: | 546 Args: |
526 timeout: (optional) Timeout per try in seconds. | 547 timeout: (optional) Timeout per try in seconds. |
527 retries: (optional) Number of retries to attempt. | 548 retries: (optional) Number of retries to attempt. |
528 """ | 549 """ |
529 output = self._RunDeviceAdbCmd(['root'], timeout, retries) | 550 output = self._RunDeviceAdbCmd(['root'], timeout, retries) |
530 if 'cannot' in output: | 551 if 'cannot' in output: |
531 raise device_errors.AdbCommandFailedError( | 552 raise device_errors.AdbCommandFailedError( |
532 ['root'], output, device_serial=self._device_serial) | 553 ['root'], output, device_serial=self._device_serial) |
OLD | NEW |