FilesExpand file tree

 master

/

tempdir.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

106 lines (81 loc) · 3.25 KB

 master

/

tempdir.py

Copy path

Top

File metadata and controls

• Code
• Blame

106 lines (81 loc) · 3.25 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

"""TemporaryDirectory class, copied from Python 3.2.

This is copied from the stdlib and will be standard in Python 3.2 and onwards.

"""

import os as _os

# This code should only be used in Python versions < 3.2, since after that we

# can rely on the stdlib itself.

try:

from tempfile import TemporaryDirectory

except ImportError:

from tempfile import mkdtemp, template

class TemporaryDirectory(object):

"""Create and return a temporary directory. This has the same

behavior as mkdtemp but can be used as a context manager. For

example:

with TemporaryDirectory() as tmpdir:

...

Upon exiting the context, the directory and everthing contained

in it are removed.

"""

def __init__(self, suffix="", prefix=template, dir=None):

self.name = mkdtemp(suffix, prefix, dir)

self._closed = False

def __enter__(self):

return self.name

def cleanup(self):

if not self._closed:

self._rmtree(self.name)

self._closed = True

def __exit__(self, exc, value, tb):

self.cleanup()

__del__ = cleanup

# XXX (ncoghlan): The following code attempts to make

# this class tolerant of the module nulling out process

# that happens during CPython interpreter shutdown

# Alas, it doesn't actually manage it. See issue #10188

_listdir = staticmethod(_os.listdir)

_path_join = staticmethod(_os.path.join)

_isdir = staticmethod(_os.path.isdir)

_remove = staticmethod(_os.remove)

_rmdir = staticmethod(_os.rmdir)

_os_error = _os.error

def _rmtree(self, path):

# Essentially a stripped down version of shutil.rmtree. We can't

# use globals because they may be None'ed out at shutdown.

for name in self._listdir(path):

fullname = self._path_join(path, name)

try:

isdir = self._isdir(fullname)

except self._os_error:

isdir = False

if isdir:

self._rmtree(fullname)

else:

try:

self._remove(fullname)

except self._os_error:

pass

try:

self._rmdir(path)

except self._os_error:

pass

class NamedFileInTemporaryDirectory(object):

def __init__(self, filename, mode='w+b', bufsize=-1, **kwds):

"""

Open a file named `filename` in a temporary directory.

This context manager is preferred over `NamedTemporaryFile` in

stdlib `tempfile` when one needs to reopen the file.

Arguments `mode` and `bufsize` are passed to `open`.

Rest of the arguments are passed to `TemporaryDirectory`.

"""

self._tmpdir = TemporaryDirectory(**kwds)

path = _os.path.join(self._tmpdir.name, filename)

self.file = open(path, mode, bufsize)

def cleanup(self):

self.file.close()

self._tmpdir.cleanup()

__del__ = cleanup

def __enter__(self):

return self.file

def __exit__(self, type, value, traceback):

self.cleanup()