Guido van Rossum | e7b146f | 2000-02-04 15:28:42 +0000 | [diff] [blame] | 1 | """Temporary files and filenames.""" |
| 2 | |
Guido van Rossum | 41f9503 | 1992-03-31 19:02:01 +0000 | [diff] [blame] | 3 | # XXX This tries to be not UNIX specific, but I don't know beans about |
| 4 | # how to choose a temp directory or filename on MS-DOS or other |
| 5 | # systems so it may have to be changed... |
Guido van Rossum | eee9498 | 1991-11-12 15:38:08 +0000 | [diff] [blame] | 6 | |
Guido van Rossum | 41f9503 | 1992-03-31 19:02:01 +0000 | [diff] [blame] | 7 | import os |
Guido van Rossum | eee9498 | 1991-11-12 15:38:08 +0000 | [diff] [blame] | 8 | |
Guido van Rossum | 41f9503 | 1992-03-31 19:02:01 +0000 | [diff] [blame] | 9 | # Parameters that the caller may set to override the defaults |
Guido van Rossum | 41f9503 | 1992-03-31 19:02:01 +0000 | [diff] [blame] | 10 | tempdir = None |
| 11 | template = None |
| 12 | |
Guido van Rossum | 41f9503 | 1992-03-31 19:02:01 +0000 | [diff] [blame] | 13 | def gettempdir(): |
Guido van Rossum | e7b146f | 2000-02-04 15:28:42 +0000 | [diff] [blame] | 14 | """Function to calculate the directory to use.""" |
Guido van Rossum | f4aaf86 | 1996-05-28 23:31:34 +0000 | [diff] [blame] | 15 | global tempdir |
Guido van Rossum | 4033ad7 | 1996-08-08 18:33:56 +0000 | [diff] [blame] | 16 | if tempdir is not None: |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 17 | return tempdir |
Guido van Rossum | 29e5f5d | 1998-04-09 14:27:57 +0000 | [diff] [blame] | 18 | try: |
| 19 | pwd = os.getcwd() |
| 20 | except (AttributeError, os.error): |
| 21 | pwd = os.curdir |
Guido van Rossum | e504c0c | 2000-08-29 14:55:03 +0000 | [diff] [blame] | 22 | attempdirs = ['/var/tmp', '/usr/tmp', '/tmp', pwd] |
Guido van Rossum | 3e065ad | 1996-08-20 20:38:59 +0000 | [diff] [blame] | 23 | if os.name == 'nt': |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 24 | attempdirs.insert(0, 'C:\\TEMP') |
| 25 | attempdirs.insert(0, '\\TEMP') |
Guido van Rossum | f4f756c | 1997-04-11 19:00:53 +0000 | [diff] [blame] | 26 | elif os.name == 'mac': |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 27 | import macfs, MACFS |
| 28 | try: |
| 29 | refnum, dirid = macfs.FindFolder(MACFS.kOnSystemDisk, |
Guido van Rossum | 57a0661 | 1998-04-28 16:03:34 +0000 | [diff] [blame] | 30 | MACFS.kTemporaryFolderType, 1) |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 31 | dirname = macfs.FSSpec((refnum, dirid, '')).as_pathname() |
| 32 | attempdirs.insert(0, dirname) |
| 33 | except macfs.error: |
| 34 | pass |
Guido van Rossum | ca54982 | 1997-08-12 18:00:12 +0000 | [diff] [blame] | 35 | for envname in 'TMPDIR', 'TEMP', 'TMP': |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 36 | if os.environ.has_key(envname): |
| 37 | attempdirs.insert(0, os.environ[envname]) |
Guido van Rossum | 3e065ad | 1996-08-20 20:38:59 +0000 | [diff] [blame] | 38 | testfile = gettempprefix() + 'test' |
Guido van Rossum | f4aaf86 | 1996-05-28 23:31:34 +0000 | [diff] [blame] | 39 | for dir in attempdirs: |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 40 | try: |
Guido van Rossum | 00f09b3 | 2000-04-24 13:28:02 +0000 | [diff] [blame] | 41 | filename = os.path.join(dir, testfile) |
| 42 | if os.name == 'posix': |
| 43 | try: |
| 44 | fd = os.open(filename, os.O_RDWR|os.O_CREAT|os.O_EXCL, 0700) |
| 45 | except OSError: |
| 46 | pass |
| 47 | else: |
| 48 | fp = os.fdopen(fd, 'w') |
| 49 | fp.write('blat') |
| 50 | fp.close() |
| 51 | os.unlink(filename) |
| 52 | del fp, fd |
| 53 | tempdir = dir |
| 54 | break |
| 55 | else: |
| 56 | fp = open(filename, 'w') |
| 57 | fp.write('blat') |
| 58 | fp.close() |
| 59 | os.unlink(filename) |
| 60 | tempdir = dir |
| 61 | break |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 62 | except IOError: |
| 63 | pass |
Guido van Rossum | f4aaf86 | 1996-05-28 23:31:34 +0000 | [diff] [blame] | 64 | if tempdir is None: |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 65 | msg = "Can't find a usable temporary directory amongst " + `attempdirs` |
| 66 | raise IOError, msg |
Guido van Rossum | f4aaf86 | 1996-05-28 23:31:34 +0000 | [diff] [blame] | 67 | return tempdir |
Guido van Rossum | 41f9503 | 1992-03-31 19:02:01 +0000 | [diff] [blame] | 68 | |
| 69 | |
Guido van Rossum | b0e5718 | 1998-10-14 20:27:05 +0000 | [diff] [blame] | 70 | _pid = None |
| 71 | |
Guido van Rossum | 41f9503 | 1992-03-31 19:02:01 +0000 | [diff] [blame] | 72 | def gettempprefix(): |
Guido van Rossum | e7b146f | 2000-02-04 15:28:42 +0000 | [diff] [blame] | 73 | """Function to calculate a prefix of the filename to use.""" |
Guido van Rossum | b0e5718 | 1998-10-14 20:27:05 +0000 | [diff] [blame] | 74 | global template, _pid |
| 75 | if os.name == 'posix' and _pid and _pid != os.getpid(): |
| 76 | # Our pid changed; we must have forked -- zap the template |
| 77 | template = None |
| 78 | if template is None: |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 79 | if os.name == 'posix': |
Guido van Rossum | b0e5718 | 1998-10-14 20:27:05 +0000 | [diff] [blame] | 80 | _pid = os.getpid() |
| 81 | template = '@' + `_pid` + '.' |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 82 | elif os.name == 'nt': |
| 83 | template = '~' + `os.getpid()` + '-' |
| 84 | elif os.name == 'mac': |
| 85 | template = 'Python-Tmp-' |
| 86 | else: |
| 87 | template = 'tmp' # XXX might choose a better one |
| 88 | return template |
Guido van Rossum | cff3454 | 1992-01-14 18:31:56 +0000 | [diff] [blame] | 89 | |
Guido van Rossum | eee9498 | 1991-11-12 15:38:08 +0000 | [diff] [blame] | 90 | |
Guido van Rossum | b8c42c9 | 1997-12-19 04:29:50 +0000 | [diff] [blame] | 91 | def mktemp(suffix=""): |
Guido van Rossum | e7b146f | 2000-02-04 15:28:42 +0000 | [diff] [blame] | 92 | """User-callable function to return a unique temporary file name.""" |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 93 | dir = gettempdir() |
| 94 | pre = gettempprefix() |
| 95 | while 1: |
Tim Peters | 1baa22a | 2001-01-12 10:02:46 +0000 | [diff] [blame^] | 96 | i = _counter.get_next() |
| 97 | file = os.path.join(dir, pre + str(i) + suffix) |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 98 | if not os.path.exists(file): |
| 99 | return file |
Guido van Rossum | ca54982 | 1997-08-12 18:00:12 +0000 | [diff] [blame] | 100 | |
| 101 | |
| 102 | class TemporaryFileWrapper: |
| 103 | """Temporary file wrapper |
| 104 | |
| 105 | This class provides a wrapper around files opened for temporary use. |
| 106 | In particular, it seeks to automatically remove the file when it is |
| 107 | no longer needed. |
| 108 | """ |
| 109 | def __init__(self, file, path): |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 110 | self.file = file |
| 111 | self.path = path |
Guido van Rossum | ca54982 | 1997-08-12 18:00:12 +0000 | [diff] [blame] | 112 | |
| 113 | def close(self): |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 114 | self.file.close() |
| 115 | os.unlink(self.path) |
Guido van Rossum | ca54982 | 1997-08-12 18:00:12 +0000 | [diff] [blame] | 116 | |
| 117 | def __del__(self): |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 118 | try: self.close() |
| 119 | except: pass |
Guido van Rossum | ca54982 | 1997-08-12 18:00:12 +0000 | [diff] [blame] | 120 | |
| 121 | def __getattr__(self, name): |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 122 | file = self.__dict__['file'] |
| 123 | a = getattr(file, name) |
Guido van Rossum | 6b708d5 | 1999-06-01 18:55:36 +0000 | [diff] [blame] | 124 | if type(a) != type(0): |
| 125 | setattr(self, name, a) |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 126 | return a |
Guido van Rossum | ca54982 | 1997-08-12 18:00:12 +0000 | [diff] [blame] | 127 | |
| 128 | |
Guido van Rossum | b8c42c9 | 1997-12-19 04:29:50 +0000 | [diff] [blame] | 129 | def TemporaryFile(mode='w+b', bufsize=-1, suffix=""): |
Guido van Rossum | e7b146f | 2000-02-04 15:28:42 +0000 | [diff] [blame] | 130 | """Create and return a temporary file (opened read-write by default).""" |
Guido van Rossum | b8c42c9 | 1997-12-19 04:29:50 +0000 | [diff] [blame] | 131 | name = mktemp(suffix) |
Guido van Rossum | dce3d55 | 1998-10-24 01:34:45 +0000 | [diff] [blame] | 132 | if os.name == 'posix': |
| 133 | # Unix -- be very careful |
| 134 | fd = os.open(name, os.O_RDWR|os.O_CREAT|os.O_EXCL, 0700) |
Guido van Rossum | 2457fc2 | 1998-10-24 15:02:59 +0000 | [diff] [blame] | 135 | try: |
| 136 | os.unlink(name) |
| 137 | return os.fdopen(fd, mode, bufsize) |
| 138 | except: |
| 139 | os.close(fd) |
| 140 | raise |
Guido van Rossum | ca54982 | 1997-08-12 18:00:12 +0000 | [diff] [blame] | 141 | else: |
Guido van Rossum | dce3d55 | 1998-10-24 01:34:45 +0000 | [diff] [blame] | 142 | # Non-unix -- can't unlink file that's still open, use wrapper |
| 143 | file = open(name, mode, bufsize) |
| 144 | return TemporaryFileWrapper(file, name) |
Tim Peters | 1baa22a | 2001-01-12 10:02:46 +0000 | [diff] [blame^] | 145 | |
| 146 | # In order to generate unique names, mktemp() uses _counter.get_next(). |
| 147 | # This returns a unique integer on each call, in a threadsafe way (i.e., |
| 148 | # multiple threads will never see the same integer). The integer will |
| 149 | # usually be a Python int, but if _counter.get_next() is called often |
| 150 | # enough, it will become a Python long. |
| 151 | # Note that the only name that survives this next block of code |
| 152 | # is "_counter". |
| 153 | |
| 154 | class _ThreadSafeCounter: |
| 155 | def __init__(self, mutex, initialvalue=0): |
| 156 | self.mutex = mutex |
| 157 | self.i = initialvalue |
| 158 | |
| 159 | def get_next(self): |
| 160 | self.mutex.acquire() |
| 161 | result = self.i |
| 162 | try: |
| 163 | newi = result + 1 |
| 164 | except OverflowError: |
| 165 | newi = long(result) + 1 |
| 166 | self.i = newi |
| 167 | self.mutex.release() |
| 168 | return result |
| 169 | |
| 170 | try: |
| 171 | import thread |
| 172 | |
| 173 | except ImportError: |
| 174 | class _DummyMutex: |
| 175 | def acquire(self): |
| 176 | pass |
| 177 | |
| 178 | release = acquire |
| 179 | |
| 180 | _counter = _ThreadSafeCounter(_DummyMutex()) |
| 181 | del _DummyMutex |
| 182 | |
| 183 | else: |
| 184 | _counter = _ThreadSafeCounter(thread.allocate_lock()) |
| 185 | del thread |
| 186 | |
| 187 | del _ThreadSafeCounter |