26b3a40133
* INCR-108: Run python-modernize on openedx/core/lib * Fixed urlencode import * Fixed diff quality warnings and six.string_types error * Fixed pickle import * Fixed iteritems import * Fixed ungrouped imports * Fixed six.moves import issues
40 lines
1.0 KiB
Python
40 lines
1.0 KiB
Python
"""Make temporary directories nicely."""
|
|
|
|
from __future__ import absolute_import
|
|
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 create_symlink(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(delete_symlink, dest)
|
|
|
|
|
|
def delete_symlink(link_path):
|
|
"""
|
|
Removes symbolic link for
|
|
:param link_path:
|
|
"""
|
|
if os.path.exists(link_path):
|
|
os.remove(link_path)
|