Add new optional parameter 'suffix' (default ''), which is appended to
the temporary file name. Also some minor formatting of Jim F's code.
diff --git a/Lib/tempfile.py b/Lib/tempfile.py
index d36898b..5b4e388 100644
--- a/Lib/tempfile.py
+++ b/Lib/tempfile.py
@@ -27,7 +27,8 @@
elif os.name == 'mac':
import macfs, MACFS
try:
- refnum, dirid = macfs.FindFolder(MACFS.kOnSystemDisk, MACFS.kTemporaryFolderType, 0)
+ refnum, dirid = macfs.FindFolder(MACFS.kOnSystemDisk,
+ MACFS.kTemporaryFolderType, 0)
dirname = macfs.FSSpec((refnum, dirid, '')).as_pathname()
attempdirs.insert(0, dirname)
except macfs.error:
@@ -76,13 +77,13 @@
# User-callable function to return a unique temporary file name
-def mktemp():
+def mktemp(suffix=""):
global counter
dir = gettempdir()
pre = gettempprefix()
while 1:
counter = counter + 1
- file = os.path.join(dir, pre + `counter`)
+ file = os.path.join(dir, pre + `counter` + suffix)
if not os.path.exists(file):
return file
@@ -95,8 +96,8 @@
no longer needed.
"""
def __init__(self, file, path):
- self.file=file
- self.path=path
+ self.file = file
+ self.path = path
def close(self):
self.file.close()
@@ -107,15 +108,15 @@
except: pass
def __getattr__(self, name):
- file=self.__dict__['file']
- a=getattr(file, name)
+ file = self.__dict__['file']
+ a = getattr(file, name)
setattr(self, name, a)
return a
-def TemporaryFile(mode='w+b', bufsize=-1):
- name=mktemp()
- file=open(name,mode,bufsize)
+def TemporaryFile(mode='w+b', bufsize=-1, suffix=""):
+ name = mktemp(suffix)
+ file = open(name, mode, bufsize)
try:
os.unlink(name)
except os.error: