[Telemetry] Implement dirty profile generation

tools/telemetry/telemetry/core/chrome/desktop_browser_backend.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.
 import glob
 import heapq
 import logging
 import os
 import subprocess as subprocess
 import shutil
 import sys
 import tempfile
 import time
 
 from telemetry.core import util
 from telemetry.core.backends import browser_backend
 from telemetry.core.backends.chrome import chrome_browser_backend
 
 class DesktopBrowserBackend(chrome_browser_backend.ChromeBrowserBackend):
   """The backend for controlling a locally-executed browser instance, on Linux,
   Mac or Windows.
   """
   def __init__(self, options, executable, flash_path, is_content_shell,
-               browser_directory, delete_profile_dir_after_run=True):
+               browser_directory, output_profile_path=None):
     super(DesktopBrowserBackend, self).__init__(
         is_content_shell=is_content_shell,
         supports_extensions=not is_content_shell,
         options=options)
 
     # Initialize fields so that an explosion during init doesn't break in Close.
     self._proc = None
     self._tmp_profile_dir = None
     self._tmp_output_file = None
 
     self._executable = executable
     if not self._executable:
       raise Exception('Cannot create browser, no executable found!')
 
     self._flash_path = flash_path
     if self._flash_path and not os.path.exists(self._flash_path):
       logging.warning(('Could not find flash at %s. Running without flash.\n\n'
                        'To fix this see http://go/read-src-internal') %
                       self._flash_path)
       self._flash_path = None
 
     if len(options.extensions_to_load) > 0 and is_content_shell:
       raise browser_backend.ExtensionsNotSupportedException(
           'Content shell does not support extensions.')
 
     self._browser_directory = browser_directory
     self._port = util.GetAvailableLocalPort()
     self._profile_dir = None
     self._supports_net_benchmarking = True
-    self._delete_profile_dir_after_run = delete_profile_dir_after_run
+    self._output_profile_path = output_profile_path
     self._tmp_minidump_dir = tempfile.mkdtemp()
 
     self._SetupProfile()
 
   def _SetupProfile(self):
     if not self.options.dont_override_profile:
       self._tmp_profile_dir = tempfile.mkdtemp()
       profile_dir = self._profile_dir or self.options.profile_dir
       if profile_dir:
         if self.is_content_shell:
@@ skipping 150 matching lines @@
       # Kill it.
       if not IsClosed():
         self._proc.kill()
         try:
           util.WaitFor(IsClosed, timeout=5)
           self._proc = None
         except util.TimeoutException:
           self._proc = None
           raise Exception('Could not shutdown the browser.')
 
-    if self._delete_profile_dir_after_run and \
-        self._tmp_profile_dir and os.path.exists(self._tmp_profile_dir):
+    # Copy profile aside if requested.
+    if self._output_profile_path:
+      if not (self._tmp_profile_dir and os.path.exists(self._tmp_profile_dir)):
+        raise Exception("No profile directory generated by Chrome: '%s'." %
+            self._tmp_profile_dir)
+      shutil.move(self._tmp_profile_dir, self._output_profile_path)
+      logging.info("Generated profile moved teporarily to '%s'." %
+          self._output_profile_path)
+
+    if self._tmp_profile_dir and os.path.exists(self._tmp_profile_dir):
       shutil.rmtree(self._tmp_profile_dir, ignore_errors=True)
       self._tmp_profile_dir = None
 
     if self._tmp_output_file:
       self._tmp_output_file.close()
       self._tmp_output_file = None
 
   def CreateForwarder(self, *port_pairs):
     return browser_backend.DoNothingForwarder(*port_pairs)