mirror of
https://github.com/tahoe-lafs/tahoe-lafs.git
synced 2025-04-07 10:56:49 +00:00
parent
61e2ae4aa8
commit
a9b152780e
@ -431,18 +431,6 @@ class FileUtil(ReallyEqualMixin, unittest.TestCase):
|
||||
fileutil.write_atomically(fn, "two", mode="") # non-binary
|
||||
self.failUnlessEqual(fileutil.read(fn), "two")
|
||||
|
||||
def test_NamedTemporaryDirectory(self):
|
||||
basedir = "util/FileUtil/test_NamedTemporaryDirectory"
|
||||
fileutil.make_dirs(basedir)
|
||||
td = fileutil.NamedTemporaryDirectory(dir=basedir)
|
||||
name = td.name
|
||||
self.failUnless(basedir in name)
|
||||
self.failUnless(basedir in repr(td))
|
||||
self.failUnless(os.path.isdir(name))
|
||||
del td
|
||||
# it is conceivable that we need to force gc here, but I'm not sure
|
||||
self.failIf(os.path.isdir(name))
|
||||
|
||||
def test_rename(self):
|
||||
basedir = "util/FileUtil/test_rename"
|
||||
fileutil.make_dirs(basedir)
|
||||
|
@ -87,34 +87,6 @@ class ReopenableNamedTemporaryFile:
|
||||
def shutdown(self):
|
||||
remove(self.name)
|
||||
|
||||
class NamedTemporaryDirectory:
|
||||
"""
|
||||
This calls tempfile.mkdtemp(), stores the name of the dir in
|
||||
self.name, and rmrf's the dir when it gets garbage collected or
|
||||
"shutdown()".
|
||||
"""
|
||||
def __init__(self, cleanup=True, *args, **kwargs):
|
||||
""" If cleanup, then the directory will be rmrf'ed when the object is shutdown. """
|
||||
self.cleanup = cleanup
|
||||
self.name = tempfile.mkdtemp(*args, **kwargs)
|
||||
|
||||
def __repr__(self):
|
||||
return "<%s instance at %x %s>" % (self.__class__.__name__, id(self), self.name)
|
||||
|
||||
def __str__(self):
|
||||
return self.__repr__()
|
||||
|
||||
def __del__(self):
|
||||
try:
|
||||
self.shutdown()
|
||||
except:
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
|
||||
def shutdown(self):
|
||||
if self.cleanup and hasattr(self, 'name'):
|
||||
rm_dir(self.name)
|
||||
|
||||
class EncryptedTemporaryFile:
|
||||
# not implemented: next, readline, readlines, xreadlines, writelines
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user