Fix screenshot.py

build/android/pylib/android_commands.py

 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """Provides an interface to communicate with the device via the adb command.
 
 Assumes adb binary is currently on system path.
 """
 # pylint: disable-all
 
 import collections
 import datetime
 import inspect
 import json
 import logging
 import os
 import re
 import shlex
 import signal
 import subprocess
 import sys
 import tempfile
 import time
 
 import cmd_helper
 import constants
 import screenshot
 import system_properties
 from utils import host_utils
+from device import device_utils
 
 try:
   from pylib import pexpect
 except ImportError:
   pexpect = None
 
 sys.path.append(os.path.join(
     constants.DIR_SOURCE_ROOT, 'third_party', 'android_testrunner'))
 import adb_interface
 import am_instrument_parser
@@ skipping 1674 matching lines @@
   def TakeScreenshot(self, host_file):
     """Saves a screenshot image to |host_file| on the host.
 
     Args:
       host_file: Absolute path to the image file to store on the host or None to
                  use an autogenerated file name.
 
     Returns:
       Resulting host file name of the screenshot.
     """
-    return screenshot.TakeScreenshot(self, host_file)
+    return screenshot.TakeScreenshot(device_utils.DeviceUtils(self), host_file)
 
   def PullFileFromDevice(self, device_file, host_file):
     """Download |device_file| on the device from to |host_file| on the host.
 
     Args:
       device_file: Absolute path to the file to retrieve from the device.
       host_file: Absolute path to the file to store on the host.
     """
     assert self._adb.Pull(device_file, host_file)
     assert os.path.exists(host_file)
@@ skipping 172 matching lines @@
   """
   def __init__(self, output):
     self._output = output
 
   def write(self, data):
     data = data.replace('\r\r\n', '\n')
     self._output.write(data)
 
   def flush(self):
     self._output.flush()