kitchen_run: initial CL

scripts/slave/robust_tempdir.py

 # Copyright 2016 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 logging
 import os
 import sys
 import tempfile
 
 # Install Infra build environment.
@@ skipping 25 matching lines @@
     Args:
       prefix (str): prefix to use for the temporary directories
       leak (bool): if True, do not remove temporary directories
                    on exit
     """
 
     self._tempdirs = []
     self._prefix = prefix
     self._leak = leak
 
-  def cleanup(self, path):
-    self._tempdirs.append(path)
+  def cleanup(self, base=None):
+    """Explicitly remove ALL temporary directories under "<base>/<prefix>".
+
+    This can be used e.g. to reduce chances of running out of disk space
+    when temporary directories are leaked.
+    """
+    base = base or tempfile.gettempdir()
+    path = os.path.join(base, self._prefix)
+    try:
+      if os.path.isdir(path):
+        LOGGER.info('Cleaning up temporary directory [%s].', path)
+        chromium_utils.RemoveDirectory(path)
+    except BaseException:
+      LOGGER.exception('Failed to clean up temporary directory [%s].', path)
 
   def tempdir(self, base=None):
     """Creates a temporary working directory and yields it.
 
     This creates two levels of directory:
       <base>/<prefix>
       <base>/<prefix>/tmpFOO
 
     On termination, the entire "<base>/<prefix>" directory is deleted,
     removing the subdirectory created by this instance as well as cleaning up
     any other temporary subdirectories leaked by previous executions.
 
     Args:
       base (str/None): The directory under which the tempdir should be created.
           If None, the default temporary directory root will be used.
     """
     base = base or tempfile.gettempdir()
-    # TODO(phajdan.jr): Clean up leaked directories at the beginning.
-    # Doing so should automatically prevent more out-of-disk-space scenarios.
     basedir = _ensure_directory(base, self._prefix)
-    self.cleanup(basedir)
+    self._tempdirs.append(basedir)
     tdir = tempfile.mkdtemp(dir=basedir)
     return tdir
 
   def __enter__(self):
     return self
 
   def __exit__(self, _et, _ev, _tb):
     self.close()
 
   def close(self):
     if self._leak:
       LOGGER.warning('Leaking temporary paths: %s', self._tempdirs)
     else:
       for path in reversed(self._tempdirs):
         try:
           if os.path.isdir(path):
             LOGGER.debug('Cleaning up temporary directory [%s].', path)
             chromium_utils.RemoveDirectory(path)
         except BaseException:
           LOGGER.exception('Failed to clean up temporary directory [%s].',
                            path)
     del(self._tempdirs[:])