OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # | 2 # |
3 # Copyright (c) 2013 The Chromium Authors. All rights reserved. | 3 # Copyright (c) 2013 The Chromium Authors. All rights reserved. |
4 # Use of this source code is governed by a BSD-style license that can be | 4 # Use of this source code is governed by a BSD-style license that can be |
5 # found in the LICENSE file. | 5 # found in the LICENSE file. |
6 | 6 |
7 """Provisions Android devices with settings required for bots. | 7 """Provisions Android devices with settings required for bots. |
8 | 8 |
9 Usage: | 9 Usage: |
10 ./provision_devices.py [-d <device serial number>] | 10 ./provision_devices.py [-d <device serial number>] |
(...skipping 13 matching lines...) Expand all Loading... |
24 from pylib.device import device_blacklist | 24 from pylib.device import device_blacklist |
25 from pylib.device import device_errors | 25 from pylib.device import device_errors |
26 from pylib.device import device_utils | 26 from pylib.device import device_utils |
27 from pylib.utils import run_tests_helper | 27 from pylib.utils import run_tests_helper |
28 | 28 |
29 sys.path.append(os.path.join(constants.DIR_SOURCE_ROOT, | 29 sys.path.append(os.path.join(constants.DIR_SOURCE_ROOT, |
30 'third_party', 'android_testrunner')) | 30 'third_party', 'android_testrunner')) |
31 import errors | 31 import errors |
32 | 32 |
33 def KillHostHeartbeat(): | 33 def KillHostHeartbeat(): |
34 ps = subprocess.Popen(['ps', 'aux'], stdout = subprocess.PIPE) | 34 ps = subprocess.Popen(['ps', 'aux'], stdout=subprocess.PIPE) |
35 stdout, _ = ps.communicate() | 35 stdout, _ = ps.communicate() |
36 matches = re.findall('\\n.*host_heartbeat.*', stdout) | 36 matches = re.findall('\\n.*host_heartbeat.*', stdout) |
37 for match in matches: | 37 for match in matches: |
38 logging.info('An instance of host heart beart running... will kill') | 38 logging.info('An instance of host heart beart running... will kill') |
39 pid = re.findall('(\S+)', match)[1] | 39 pid = re.findall(r'(\S+)', match)[1] |
40 subprocess.call(['kill', str(pid)]) | 40 subprocess.call(['kill', str(pid)]) |
41 | 41 |
42 | 42 |
43 def LaunchHostHeartbeat(): | 43 def LaunchHostHeartbeat(): |
44 # Kill if existing host_heartbeat | 44 # Kill if existing host_heartbeat |
45 KillHostHeartbeat() | 45 KillHostHeartbeat() |
46 # Launch a new host_heartbeat | 46 # Launch a new host_heartbeat |
47 logging.info('Spawning host heartbeat...') | 47 logging.info('Spawning host heartbeat...') |
48 subprocess.Popen([os.path.join(constants.DIR_SOURCE_ROOT, | 48 subprocess.Popen([os.path.join(constants.DIR_SOURCE_ROOT, |
49 'build/android/host_heartbeat.py')]) | 49 'build/android/host_heartbeat.py')]) |
(...skipping 133 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
183 if not device.old_interface.IsDeviceCharging(): | 183 if not device.old_interface.IsDeviceCharging(): |
184 if device.old_interface.CanControlUsbCharging(): | 184 if device.old_interface.CanControlUsbCharging(): |
185 device.old_interface.EnableUsbCharging() | 185 device.old_interface.EnableUsbCharging() |
186 else: | 186 else: |
187 logging.error('Device is not charging') | 187 logging.error('Device is not charging') |
188 break | 188 break |
189 logging.info('Waiting for device to charge. Current level=%s', | 189 logging.info('Waiting for device to charge. Current level=%s', |
190 battery_info.get('level', 0)) | 190 battery_info.get('level', 0)) |
191 time.sleep(60) | 191 time.sleep(60) |
192 battery_info = device.old_interface.GetBatteryInfo() | 192 battery_info = device.old_interface.GetBatteryInfo() |
193 # TODO(jbudorick): Tune the timeout per OS version. | 193 if not options.skip_wipe: |
194 device.Reboot(True, timeout=600, retries=0) | 194 # TODO(jbudorick): Tune the timeout per OS version. |
| 195 device.Reboot(True, timeout=600, retries=0) |
195 device.RunShellCommand('date -s %s' % time.strftime('%Y%m%d.%H%M%S', | 196 device.RunShellCommand('date -s %s' % time.strftime('%Y%m%d.%H%M%S', |
196 time.gmtime()), | 197 time.gmtime()), |
197 as_root=True) | 198 as_root=True) |
198 props = device.RunShellCommand('getprop') | 199 props = device.RunShellCommand('getprop') |
199 for prop in props: | 200 for prop in props: |
200 logging.info(' %s' % prop) | 201 logging.info(' %s' % prop) |
201 if options.auto_reconnect: | 202 if options.auto_reconnect: |
202 PushAndLaunchAdbReboot(device, options.target) | 203 PushAndLaunchAdbReboot(device, options.target) |
203 except (errors.WaitForResponseTimedOutError, | 204 except (errors.WaitForResponseTimedOutError, |
204 device_errors.CommandTimeoutError): | 205 device_errors.CommandTimeoutError): |
205 logging.info('Timed out waiting for device %s. Adding to blacklist.', | 206 logging.info('Timed out waiting for device %s. Adding to blacklist.', |
206 str(device)) | 207 str(device)) |
207 # Device black list is reset by bb_device_status_check.py per build. | 208 # Device black list is reset by bb_device_status_check.py per build. |
208 device_blacklist.ExtendBlacklist([str(device)]) | 209 device_blacklist.ExtendBlacklist([str(device)]) |
209 except (device_errors.CommandFailedError): | 210 except device_errors.CommandFailedError: |
210 logging.exception('Failed to provision device %s. Adding to blacklist.', | 211 logging.exception('Failed to provision device %s. Adding to blacklist.', |
211 str(device)) | 212 str(device)) |
212 device_blacklist.ExtendBlacklist([str(device)]) | 213 device_blacklist.ExtendBlacklist([str(device)]) |
213 | 214 |
214 | 215 |
215 def ProvisionDevices(options): | 216 def ProvisionDevices(options): |
216 is_perf = 'perf' in os.environ.get('BUILDBOT_BUILDERNAME', '').lower() | 217 is_perf = 'perf' in os.environ.get('BUILDBOT_BUILDERNAME', '').lower() |
217 if options.device is not None: | 218 if options.device is not None: |
218 devices = [options.device] | 219 devices = [options.device] |
219 else: | 220 else: |
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
251 | 252 |
252 if args: | 253 if args: |
253 print >> sys.stderr, 'Unused args %s' % args | 254 print >> sys.stderr, 'Unused args %s' % args |
254 return 1 | 255 return 1 |
255 | 256 |
256 return ProvisionDevices(options) | 257 return ProvisionDevices(options) |
257 | 258 |
258 | 259 |
259 if __name__ == '__main__': | 260 if __name__ == '__main__': |
260 sys.exit(main(sys.argv)) | 261 sys.exit(main(sys.argv)) |
OLD | NEW |