954dae584a
ziafazal: improvements need for multi-tenancy ziafazal: fixed broken tests ziafazal: no need to add setting in test.py ziafazal: added hostname validation ziafazal: changes after feedback from mattdrayer ziafazal: fixed branding and microsite broken tests ziafazal: make STATICFILES_DIRS to list ziafazal: added theme directory to mako lookup for tests ziafazal: added more protection in test_util saleem-latif: Enable SCSS Overrides for Comprehensive Theming saleem-latif: Incoporate feedback changes, Correct test failures, add tests and enable theming for django templates saleem-latif: Correct errors in python tests mattdrayer: Fix invalid release reference mattdrayer: Update django-wiki reference to latest release
39 lines
992 B
Python
39 lines
992 B
Python
"""Make temporary directories nicely."""
|
|
|
|
import atexit
|
|
import os.path
|
|
import shutil
|
|
import tempfile
|
|
|
|
|
|
def mkdtemp_clean(suffix="", prefix="tmp", dir=None): # pylint: disable=redefined-builtin
|
|
"""Just like mkdtemp, but the directory will be deleted when the process ends."""
|
|
the_dir = tempfile.mkdtemp(suffix=suffix, prefix=prefix, dir=dir)
|
|
atexit.register(cleanup_tempdir, the_dir)
|
|
return the_dir
|
|
|
|
|
|
def cleanup_tempdir(the_dir):
|
|
"""Called on process exit to remove a temp directory."""
|
|
if os.path.exists(the_dir):
|
|
shutil.rmtree(the_dir)
|
|
|
|
|
|
def mksym_link(src, dest):
|
|
"""
|
|
Creates a symbolic link which will be deleted when the process ends.
|
|
:param src: path to source
|
|
:param dest: path to destination
|
|
"""
|
|
os.symlink(src, dest)
|
|
atexit.register(cleanup_symlink, dest)
|
|
|
|
|
|
def cleanup_symlink(link_path):
|
|
"""
|
|
Removes symbolic link for
|
|
:param link_path:
|
|
"""
|
|
if os.path.exists(link_path):
|
|
os.remove(link_path)
|