tempfile: Add initial tempfile implementation.

With TemporaryDirectory class and mkdtemp function.

Author: BrianPugh

python-stdlib/tempfile/tempfile.py

@@ -0,0 +1,59 @@
+import errno
+import os
+import random
+import shutil
+
+_ascii_letters = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
+
+
+def _get_candidate_name(size=8):
+    return "".join(random.choice(_ascii_letters) for _ in range(size))
+
+
+def _sanitize_inputs(suffix, prefix, dir):
+    if dir is None:
+        dir = "/tmp"
+    if suffix is None:
+        suffix = ""
+    if prefix is None:
+        prefix = ""
+    return suffix, prefix, dir
+
+
+def _try(action, *args, **kwargs):
+    try:
+        action(*args, **kwargs)
+        return True
+    except OSError as e:
+        if e.errno != errno.EEXIST:
+            raise e
+    return False
+
+
+def mkdtemp(suffix=None, prefix=None, dir=None):
+    suffix, prefix, dir = _sanitize_inputs(suffix, prefix, dir)
+
+    _try(os.mkdir, dir)
+
+    while True:
+        name = _get_candidate_name()
+        file = os.path.join(dir, prefix + name + suffix)
+        if _try(os.mkdir, file):
+            return file
+
+
+class TemporaryDirectory:
+    def __init__(self, suffix=None, prefix=None, dir=None):
+        self.name = mkdtemp(suffix, prefix, dir)
+
+    def __repr__(self):
+        return "<{} {!r}>".format(self.__class__.__name__, self.name)
+
+    def __enter__(self):
+        return self.name
+
+    def __exit__(self, exc, value, tb):
+        self.cleanup()
+
+    def cleanup(self):
+        _try(shutil.rmtree, self.name)