From aa14fb15fae6447ae644b85ac7de0190ff643322 Mon Sep 17 00:00:00 2001
From: deadc0de6 <info@deadc0de.ch>
Date: Tue, 3 Nov 2020 22:22:44 +0100
Subject: [PATCH] implement DOTDROP_TMPDIR env for #268

---
 dotdrop/utils.py | 29 ++++++++++++++++++++++++++---
 1 file changed, 26 insertions(+), 3 deletions(-)

diff --git a/dotdrop/utils.py b/dotdrop/utils.py
index 6048952..0435973 100644
--- a/dotdrop/utils.py
+++ b/dotdrop/utils.py
@@ -19,6 +19,10 @@ from dotdrop.logger import Logger
 
 LOG = Logger()
 STAR = '*'
+# the environment variable for temporary
+ENV_TEMP = 'DOTDROP_TMPDIR'
+# the temporary directory
+TMPDIR = None
 
 # files dotdrop refuses to remove
 DONOTDELETE = [
@@ -87,20 +91,39 @@ def diff(original, modified, raw=True,
 
 
 def get_tmpdir():
-    """create a temporary directory"""
+    """create and return the temporary directory"""
+    global TMPDIR
+    if TMPDIR:
+        return TMPDIR
+    t = _get_tmpdir()
+    TMPDIR = t
+    return t
+
+
+def _get_tmpdir():
+    """create the tmpdir"""
+    if ENV_TEMP in os.environ:
+        t = os.environ[ENV_TEMP]
+        t = os.path.expanduser(t)
+        t = os.path.abspath(t)
+        t = os.path.normpath(t)
+        os.makedirs(t, exist_ok=True)
+        return t
     return tempfile.mkdtemp(prefix='dotdrop-')
 
 
 def get_tmpfile():
     """create a temporary file"""
-    (_, path) = tempfile.mkstemp(prefix='dotdrop-')
+    tmpdir = get_tmpdir()
+    (_, path) = tempfile.mkstemp(prefix='dotdrop-', dir=tmpdir)
     return path
 
 
 def get_unique_tmp_name():
     """get a unique file name (not created)"""
     unique = str(uuid.uuid4())
-    return os.path.join(tempfile.gettempdir(), unique)
+    tmpdir = get_tmpdir()
+    return os.path.join(tmpdir, unique)
 
 
 def remove(path, logger=None):