Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(2741)

Unified Diff: build/android/pylib/screenshot.py

Issue 386053002: [Android] Switch to DeviceUtils versions of GetPid, TakeScreenshot, and GetIoStats. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: change GetPids and fix screenshot Created 6 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: build/android/pylib/screenshot.py
diff --git a/build/android/pylib/screenshot.py b/build/android/pylib/screenshot.py
index 390837292fa5d59532b38ba1d7b74c6d5a6af25c..4e0e7d7f95fad8d9ec361b9a5d6766ba8221f145 100644
--- a/build/android/pylib/screenshot.py
+++ b/build/android/pylib/screenshot.py
@@ -3,6 +3,7 @@
# found in the LICENSE file.
import os
+import signal
import tempfile
from pylib import cmd_helper
@@ -33,7 +34,6 @@ class VideoRecorder(object):
self._device_file = (
'%s/screen-recording.mp4' % device.GetExternalStoragePath())
self._recorder = None
- self._recorder_pids = None
self._recorder_stdout = None
self._is_started = False
@@ -53,8 +53,7 @@ class VideoRecorder(object):
self._recorder_stdout = tempfile.mkstemp()[1]
self._recorder = cmd_helper.Popen(
self._args, stdout=open(self._recorder_stdout, 'w'))
- self._recorder_pids = self._device.old_interface.ExtractPid('screenrecord')
- if not self._recorder_pids:
+ if not self._device.GetPids('screenrecord'):
raise RuntimeError('Recording failed. Is your device running Android '
'KitKat or later?')
@@ -70,21 +69,23 @@ class VideoRecorder(object):
"""Stop recording video."""
os.remove(self._recorder_stdout)
self._is_started = False
- if not self._recorder or not self._recorder_pids:
+ if not self._recorder:
return
- self._device.RunShellCommand(
- 'kill -SIGINT ' + ' '.join(self._recorder_pids))
+ self._device.KillAll('screenrecord', signum=signal.SIGINT)
self._recorder.wait()
- def Pull(self, host_file):
+ def Pull(self, host_file=None):
jbudorick 2014/07/15 15:52:57 Changes in here are to fix issues introduced in ht
"""Pull resulting video file from the device.
Args:
host_file: Path to the video file to store on the host.
+ Returns:
+ Output video file name on the host.
"""
- host_file_name = host_file or ('screen-recording-%s.mp4' %
- self._device.old_interface.GetTimestamp())
- host_file = os.path.abspath(host_file_name)
- self._device.old_interface.EnsureHostDirectory(self._host_file)
- self._device.PullFile(self._device_file, host_file_name)
+ host_file = host_file or ('screen-recording-%s.mp4' %
+ self._device.old_interface.GetTimestamp())
+ host_file = os.path.abspath(host_file)
+ self._device.old_interface.EnsureHostDirectory(host_file)
+ self._device.PullFile(self._device_file, host_file)
self._device.RunShellCommand('rm -f "%s"' % self._device_file)
+ return host_file

Powered by Google App Engine
This is Rietveld 408576698