summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorRadostin Stoyanov <rstoyanov1@gmail.com>2017-08-26 21:42:10 +0100
committerRadostin Stoyanov <rstoyanov1@gmail.com>2017-08-28 17:01:14 +0100
commit1d95bd95fd620b1f0049945c707406432f09a6db (patch)
tree91933ebbc503c6e141fb5966e754e1e65246c5fa
parent1ba2252ac9a83d54091396d88008e0249bbb7ccf (diff)
downloadvirt-bootstrap.git-1d95bd95fd620b1f0049945c707406432f09a6db.tar.gz
virt-bootstrap.git-1d95bd95fd620b1f0049945c707406432f09a6db.tar.xz
virt-bootstrap.git-1d95bd95fd620b1f0049945c707406432f09a6db.zip
Use environment variable for temporary directory
Allow the user to specify path for temporary directory used by virt-bootstrap using the `VIRTBOOTSTRAP_TMPDIR` environment variable.
-rw-r--r--src/virtBootstrap/utils.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/src/virtBootstrap/utils.py b/src/virtBootstrap/utils.py
index d5da5c1..489592c 100644
--- a/src/virtBootstrap/utils.py
+++ b/src/virtBootstrap/utils.py
@@ -52,6 +52,12 @@ else:
DEFAULT_IMG_DIR = os.environ['HOME']
DEFAULT_IMG_DIR += "/.local/share/virt-bootstrap/docker_images"
+# Set temporary directory
+tmp_dir = os.environ.get('VIRTBOOTSTRAP_TMPDIR', '/tmp')
+if not os.path.exists(tmp_dir):
+ os.makedirs(tmp_dir)
+tempfile.tempdir = tmp_dir
+
class BuildImage(object):
"""