Enable "get html from page" functionality for PyAuto....

chrome/test/pyautolib/pyauto.py

 #!/usr/bin/python
 
 # Copyright (c) 2010 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.
 
 """PyAuto: Python Interface to Chromium's Automation Proxy.
 
 PyAuto uses swig to expose Automation Proxy interfaces to Python.
 For complete documentation on the functionality available,
@@ skipping 11 matching lines @@
 
 This script can be used as an executable to fire off other scripts, similar
 to unittest.py
   python pyauto.py test_script
 """
 
 import logging
 import optparse
 import os
 import re
+import shutil
 import sys
+import tempfile
 import time
 import types
 import unittest
 import urllib
 
 
 def _LocateBinDirs():
   """Setup a few dirs where we expect to find dependency libraries."""
   script_dir = os.path.dirname(__file__)
   chrome_src = os.path.join(script_dir, os.pardir, os.pardir, os.pardir)
@@ skipping 567 matching lines @@
       pyauto_errors.JSONInterfaceError if the automation call returns an error.
     """
     cmd_dict = {
       'command': 'DisablePlugin',
       'path': path,
     }
     ret_dict = json.loads(self._SendJSONRequest(0, json.dumps(cmd_dict)))
     if ret_dict.has_key('error'):
       raise JSONInterfaceError(ret_dict['error'])
 
+  def GetTabContents(self, tab_index=0, window_index=0):
+    """Get the html contents of a tab (a la "view source").
+
+    As an implementation detail, this saves the html in a file, reads
+    the file into a buffer, then deletes it.
+
+    Args:
+      tab_index: tab index, defaults to 0.
+      window_index: window index, defaults to 0.
+    Returns:
+      html content of a page as a string.
+    """
+    tempdir = tempfile.mkdtemp()
+    filename = os.path.join(tempdir, 'content.html')
+    cmd_dict = {  # Prepare command for the json interface
+      'command': 'SaveTabContents',
+      'tab_index': tab_index,
+      'filename': filename
+    }
+    ret_dict = json.loads(self._SendJSONRequest(window_index,
+                                                json.dumps(cmd_dict)))
+    if ret_dict.has_key('error'):
+      raise JSONInterfaceError(ret_dict['error'])
+    try:
+      f = open(filename)
+      all_data = f.read()
+      f.close()
+      return all_data
+    except IOError:
+      raise
+    finally:
+      shutil.rmtree(tempdir)
+
 
 class PyUITestSuite(pyautolib.PyUITestSuiteBase, unittest.TestSuite):
   """Base TestSuite for PyAuto UI tests."""
 
   def __init__(self, args):
     pyautolib.PyUITestSuiteBase.__init__(self, args)
 
     # Figure out path to chromium binaries
     browser_dir = os.path.normpath(os.path.dirname(pyautolib.__file__))
     logging.debug('Loading pyauto libs from %s', browser_dir)
@@ skipping 213 matching lines @@
     if self._options.verbose:
       verbosity = 2
     result = unittest.TextTestRunner(verbosity=verbosity).run(pyauto_suite)
     del loaded_tests  # Need to destroy test cases before the suite
     del pyauto_suite
     sys.exit(not result.wasSuccessful())
 
 
 if __name__ == '__main__':
   Main()