add 'temp_dir' setting to GlobalConfig

Author: funbringer

testgres/backup.py

@@ -2,9 +2,11 @@
 
 import os
 import shutil
-import tempfile
 
 from six import raise_from
+from tempfile import mkdtemp
+
+from .config import testgres_config
 
 from .consts import \
     DATA_DIR, \
@@ -49,9 +51,11 @@ def __init__(self,
         if not node.status():
             raise BackupException('Node must be running')
 
+        # yapf: disable
         # Set default arguments
         username = username or default_username()
-        base_dir = base_dir or tempfile.mkdtemp(prefix=TMP_BACKUP)
+        base_dir = base_dir or mkdtemp(prefix=TMP_BACKUP,
+                                       dir=testgres_config.temp_dir)
 
         # public
         self.original_node = node
@@ -98,7 +102,8 @@ def _prepare_dir(self, destroy):
         available = not destroy
 
         if available:
-            dest_base_dir = tempfile.mkdtemp(prefix=TMP_NODE)
+            dest_base_dir = mkdtemp(prefix=TMP_NODE,
+                                    dir=testgres_config.temp_dir)
 
             data1 = os.path.join(self.base_dir, DATA_DIR)
             data2 = os.path.join(dest_base_dir, DATA_DIR)

testgres/config.py

@@ -3,9 +3,9 @@
 import atexit
 import copy
 import shutil
-import tempfile
 
 from contextlib import contextmanager
+from tempfile import mkdtemp
 
 from .consts import TMP_CACHE
 
@@ -21,6 +21,8 @@ class GlobalConfig(object):
 
         cache_pg_config:        shall we cache pg_config results?
 
+        temp_dir:               base temp dir for nodes with default 'base_dir'.
+
         use_python_logging:     use python logging configuration for all nodes.
         error_log_lines:        N of log lines to be shown in exception (0=inf).
 
@@ -37,6 +39,8 @@ class GlobalConfig(object):
 
     cache_pg_config = True
 
+    temp_dir = None
+
     use_python_logging = False
     error_log_lines = 20
 
@@ -166,5 +170,7 @@ def configure_testgres(**options):
     testgres_config.update(options)
 
 
+# yapf: disable
 # NOTE: assign initial cached dir for initdb
-testgres_config.cached_initdb_dir = tempfile.mkdtemp(prefix=TMP_CACHE)
+testgres_config.cached_initdb_dir = mkdtemp(prefix=TMP_CACHE,
+                                            dir=testgres_config.temp_dir)

testgres/node.py

@@ -5,11 +5,11 @@
 import shutil
 import six
 import subprocess
-import tempfile
 import time
 
 from enum import Enum
 from six import raise_from
+from tempfile import mkstemp, mkdtemp
 
 from .cache import cached_initdb
 
@@ -200,7 +200,8 @@ def _create_recovery_conf(self, username):
 
     def _prepare_dirs(self):
         if not self.base_dir:
-            self.base_dir = tempfile.mkdtemp(prefix=TMP_NODE)
+            self.base_dir = mkdtemp(prefix=TMP_NODE,
+                                    dir=testgres_config.temp_dir)
 
         if not os.path.exists(self.base_dir):
             os.makedirs(self.base_dir)
@@ -718,7 +719,7 @@ def dump(self, filename=None, dbname=None, username=None):
         """
 
         def tmpfile():
-            fd, fname = tempfile.mkstemp(prefix=TMP_DUMP)
+            fd, fname = mkstemp(prefix=TMP_DUMP, dir=testgres_config.temp_dir)
             os.close(fd)
             return fname