53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
|
import sys
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
__all__ = ['get_config_vars', 'get_path']
|
||
|
|
|
||
|
|
try:
|
||
|
|
# Python 2.7 or >=3.2
|
||
|
|
from sysconfig import get_config_vars, get_path
|
||
|
|
except ImportError:
|
||
|
|
from distutils.sysconfig import get_config_vars, get_python_lib
|
||
|
|
def get_path(name):
|
||
|
|
if name not in ('platlib', 'purelib'):
|
||
|
|
raise ValueError("Name must be purelib or platlib")
|
||
|
|
return get_python_lib(name=='platlib')
|
||
|
|
|
||
|
|
try:
|
||
|
|
# Python >=3.2
|
||
|
|
from tempfile import TemporaryDirectory
|
||
|
|
except ImportError:
|
||
|
|
import shutil
|
||
|
|
import tempfile
|
||
|
|
class TemporaryDirectory(object):
|
||
|
|
""""
|
||
|
|
Very simple temporary directory context manager.
|
||
|
|
Will try to delete afterward, but will also ignore OS and similar
|
||
|
|
errors on deletion.
|
||
|
|
"""
|
||
|
|
def __init__(self):
|
||
|
|
self.name = None # Handle mkdtemp raising an exception
|
||
|
|
self.name = tempfile.mkdtemp()
|
||
|
|
|
||
|
|
def __enter__(self):
|
||
|
|
return self.name
|
||
|
|
|
||
|
|
def __exit__(self, exctype, excvalue, exctrace):
|
||
|
|
try:
|
||
|
|
shutil.rmtree(self.name, True)
|
||
|
|
except OSError: #removal errors are not the only possible
|
||
|
|
pass
|
||
|
|
self.name = None
|
||
|
|
|
||
|
|
|
||
|
|
unittest_main = unittest.main
|
||
|
|
|
||
|
|
_PY31 = (3, 1) <= sys.version_info[:2] < (3, 2)
|
||
|
|
if _PY31:
|
||
|
|
# on Python 3.1, translate testRunner==None to TextTestRunner
|
||
|
|
# for compatibility with Python 2.6, 2.7, and 3.2+
|
||
|
|
def unittest_main(*args, **kwargs):
|
||
|
|
if 'testRunner' in kwargs and kwargs['testRunner'] is None:
|
||
|
|
kwargs['testRunner'] = unittest.TextTestRunner
|
||
|
|
return unittest.main(*args, **kwargs)
|