60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
import os,shutil,tempfile
|
|
import base
|
|
|
|
class Pathed(base.Base):
|
|
# Temporary files
|
|
#repos='/tmp/test_repos_091x10'
|
|
#config=repos+'/migrate.cfg'
|
|
#script='/tmp/test_migration_script.py'
|
|
|
|
_tmpdir=tempfile.mkdtemp()
|
|
|
|
@classmethod
|
|
def _tmp(cls,prefix='',suffix=''):
|
|
"""Generate a temporary file name that doesn't exist
|
|
All filenames are generated inside a temporary directory created by
|
|
tempfile.mkdtemp(); only the creating user has access to this directory.
|
|
It should be secure to return a nonexistant temp filename in this
|
|
directory, unless the user is messing with their own files.
|
|
"""
|
|
file,ret = tempfile.mkstemp(suffix,prefix,cls._tmpdir)
|
|
os.close(file)
|
|
os.remove(ret)
|
|
return ret
|
|
|
|
@classmethod
|
|
def tmp(cls,*p,**k):
|
|
return cls._tmp(*p,**k)
|
|
|
|
@classmethod
|
|
def tmp_py(cls,*p,**k):
|
|
return cls._tmp(suffix='.py',*p,**k)
|
|
|
|
@classmethod
|
|
def tmp_sql(cls,*p,**k):
|
|
return cls._tmp(suffix='.sql',*p,**k)
|
|
|
|
@classmethod
|
|
def tmp_named(cls,name):
|
|
return os.path.join(cls._tmpdir,name)
|
|
|
|
@classmethod
|
|
def tmp_repos(cls,*p,**k):
|
|
return cls._tmp(*p,**k)
|
|
|
|
@classmethod
|
|
def purge(cls,path):
|
|
"""Removes this path if it exists, in preparation for tests
|
|
Careful - all tests should take place in /tmp.
|
|
We don't want to accidentally wipe stuff out...
|
|
"""
|
|
if os.path.exists(path):
|
|
if os.path.isdir(path):
|
|
shutil.rmtree(path)
|
|
else:
|
|
os.remove(path)
|
|
if path.endswith('.py'):
|
|
pyc = path+'c'
|
|
if os.path.exists(pyc):
|
|
os.remove(pyc)
|