cpython/Lib/tempfile.py

75 lines
1.6 KiB
Python
Raw Normal View History

1991-11-12 11:38:08 -04:00
# Temporary file name allocation
#
# XXX This tries to be not UNIX specific, but I don't know beans about
# how to choose a temp directory or filename on MS-DOS or other
# systems so it may have to be changed...
1991-11-12 11:38:08 -04:00
import os
1991-11-12 11:38:08 -04:00
# Parameters that the caller may set to override the defaults
1991-11-12 11:38:08 -04:00
tempdir = None
template = None
# Function to calculate the directory to use
def gettempdir():
1996-05-28 20:31:34 -03:00
global tempdir
if tempdir is not None:
return tempdir
1996-05-28 20:31:34 -03:00
attempdirs = ['/usr/tmp', '/tmp', os.getcwd(), os.curdir]
1996-08-20 17:38:59 -03:00
if os.name == 'nt':
attempdirs.insert(0, 'C:\\TEMP')
attempdirs.insert(0, '\\TEMP')
1996-05-28 20:31:34 -03:00
if os.environ.has_key('TMPDIR'):
attempdirs.insert(0, os.environ['TMPDIR'])
1996-08-20 17:38:59 -03:00
testfile = gettempprefix() + 'test'
1996-05-28 20:31:34 -03:00
for dir in attempdirs:
try:
filename = os.path.join(dir, testfile)
fp = open(filename, 'w')
fp.write('blat')
fp.close()
os.unlink(filename)
tempdir = dir
break
except IOError:
pass
if tempdir is None:
msg = "Can't find a usable temporary directory amongst " + `attempdirs`
raise IOError, msg
return tempdir
# Function to calculate a prefix of the filename to use
def gettempprefix():
global template
if template == None:
if os.name == 'posix':
template = '@' + `os.getpid()` + '.'
else:
template = 'tmp' # XXX might choose a better one
return template
1991-11-12 11:38:08 -04:00
1991-12-26 09:10:50 -04:00
# Counter for generating unique names
1991-11-12 11:38:08 -04:00
1991-12-26 09:10:50 -04:00
counter = 0
1991-11-12 11:38:08 -04:00
# User-callable function to return a unique temporary file name
1991-11-12 11:38:08 -04:00
def mktemp():
1991-12-26 09:10:50 -04:00
global counter
dir = gettempdir()
pre = gettempprefix()
1991-11-12 11:38:08 -04:00
while 1:
counter = counter + 1
file = os.path.join(dir, pre + `counter`)
if not os.path.exists(file):
return file