Method get_file

tools/tempfiles.py:30–48

Return an object representing a RAII-like access to a temp file. The the result is a context manager object that can be used in a 'with' statement: 'with TempFiles.get_file(..) as filename:'. The file will be deleted immediately once the 'with' block is exited.

(self, suffix)

    return named_file

  def get_file(self, suffix):
    """Return an object representing a RAII-like access to a temp file.

    The the result is a context manager object that can be used in a 'with' statement:

      'with TempFiles.get_file(..) as filename:'.

    The file will be deleted immediately once the 'with' block is exited.
    """
    class TempFileObject:
      def __enter__(self_):  # noqa: DC02
        self_.file = tempfile.NamedTemporaryFile(dir=self.tmpdir, suffix=suffix, delete=False)
        self_.file.close() # NamedTemporaryFile passes out open file handles, but callers prefer filenames (and open their own handles manually if needed)
        return self_.file.name

      def __exit__(self_, _type, _value, _traceback):  # noqa: DC02
        if not self.save_debug_files:
          utils.delete_file(self_.file.name)
    return TempFileObject()

  def get_dir(self):
    """Return a named temp directory with the given prefix."""