blob: c59aca1e1890864dc8c21a97bbcbb9b104190901 [file] [log] [blame]
Georg Brandl116aa622007-08-15 14:28:22 +00001:mod:`tempfile` --- Generate temporary files and directories
2============================================================
3
Georg Brandl116aa622007-08-15 14:28:22 +00004.. module:: tempfile
5 :synopsis: Generate temporary files and directories.
6
Terry Jan Reedyfa089b92016-06-11 15:02:54 -04007.. sectionauthor:: Zack Weinberg <zack@codesourcery.com>
8
9**Source code:** :source:`Lib/tempfile.py`
Georg Brandl116aa622007-08-15 14:28:22 +000010
11.. index::
12 pair: temporary; file name
13 pair: temporary; file
14
Raymond Hettingera1993682011-01-27 01:20:32 +000015--------------
16
Robert Collins2ebdc132015-08-13 11:38:02 +120017This module creates temporary files and directories. It works on all
18supported platforms. :class:`TemporaryFile`, :class:`NamedTemporaryFile`,
19:class:`TemporaryDirectory`, and :class:`SpooledTemporaryFile` are high-level
20interfaces which provide automatic cleanup and can be used as
21context managers. :func:`mkstemp` and
22:func:`mkdtemp` are lower-level functions which require manual cleanup.
Georg Brandl116aa622007-08-15 14:28:22 +000023
Robert Collins2ebdc132015-08-13 11:38:02 +120024All the user-callable functions and constructors take additional arguments which
25allow direct control over the location and name of temporary files and
26directories. Files names used by this module include a string of
27random characters which allows those files to be securely created in
28shared temporary directories.
Christian Heimes81ee3ef2008-05-04 22:42:01 +000029To maintain backward compatibility, the argument order is somewhat odd; it
30is recommended to use keyword arguments for clarity.
Georg Brandl116aa622007-08-15 14:28:22 +000031
Nick Coghlan543af752010-10-24 11:23:25 +000032The module defines the following user-callable items:
Georg Brandl116aa622007-08-15 14:28:22 +000033
Martin Panter9b566c32015-11-07 00:32:50 +000034.. function:: TemporaryFile(mode='w+b', buffering=None, encoding=None, newline=None, suffix=None, prefix=None, dir=None)
Georg Brandl116aa622007-08-15 14:28:22 +000035
Antoine Pitrou11cb9612010-09-15 11:11:28 +000036 Return a :term:`file-like object` that can be used as a temporary storage area.
Robert Collins2ebdc132015-08-13 11:38:02 +120037 The file is created securely, using the same rules as :func:`mkstemp`. It will be destroyed as soon
Georg Brandl116aa622007-08-15 14:28:22 +000038 as it is closed (including an implicit close when the object is garbage
Robert Collins2ebdc132015-08-13 11:38:02 +120039 collected). Under Unix, the directory entry for the file is either not created at all or is removed
Christian Heimes81ee3ef2008-05-04 22:42:01 +000040 immediately after the file is created. Other platforms do not support
41 this; your code should not rely on a temporary file created using this
42 function having or not having a visible name in the file system.
Georg Brandl116aa622007-08-15 14:28:22 +000043
Robert Collins2ebdc132015-08-13 11:38:02 +120044 The resulting object can be used as a context manager (see
45 :ref:`tempfile-examples`). On completion of the context or
46 destruction of the file object the temporary file will be removed
47 from the filesystem.
48
Christian Heimes81ee3ef2008-05-04 22:42:01 +000049 The *mode* parameter defaults to ``'w+b'`` so that the file created can
50 be read and written without being closed. Binary mode is used so that it
51 behaves consistently on all platforms without regard for the data that is
Georg Brandl14dfede2010-05-21 21:12:07 +000052 stored. *buffering*, *encoding* and *newline* are interpreted as for
53 :func:`open`.
Georg Brandl116aa622007-08-15 14:28:22 +000054
Martin Panter9b566c32015-11-07 00:32:50 +000055 The *dir*, *prefix* and *suffix* parameters have the same meaning and
56 defaults as with :func:`mkstemp`.
Georg Brandl116aa622007-08-15 14:28:22 +000057
Christian Heimes7f044312008-01-06 17:05:40 +000058 The returned object is a true file object on POSIX platforms. On other
Georg Brandl502d9a52009-07-26 15:02:41 +000059 platforms, it is a file-like object whose :attr:`!file` attribute is the
Robert Collins2ebdc132015-08-13 11:38:02 +120060 underlying true file object.
Christian Heimes7f044312008-01-06 17:05:40 +000061
Victor Stinnerd967fc92014-06-05 14:27:45 +020062 The :py:data:`os.O_TMPFILE` flag is used if it is available and works
Robert Collins2ebdc132015-08-13 11:38:02 +120063 (Linux-specific, requires Linux kernel 3.11 or later).
Victor Stinnerd967fc92014-06-05 14:27:45 +020064
65 .. versionchanged:: 3.5
66
67 The :py:data:`os.O_TMPFILE` flag is now used if available.
68
Georg Brandl116aa622007-08-15 14:28:22 +000069
Martin Panter9b566c32015-11-07 00:32:50 +000070.. function:: NamedTemporaryFile(mode='w+b', buffering=None, encoding=None, newline=None, suffix=None, prefix=None, dir=None, delete=True)
Georg Brandl116aa622007-08-15 14:28:22 +000071
Christian Heimes81ee3ef2008-05-04 22:42:01 +000072 This function operates exactly as :func:`TemporaryFile` does, except that
73 the file is guaranteed to have a visible name in the file system (on
74 Unix, the directory entry is not unlinked). That name can be retrieved
Martin Panter1f0e1f32016-02-22 10:10:00 +000075 from the :attr:`name` attribute of the returned
76 file-like object. Whether the name can be
Christian Heimes81ee3ef2008-05-04 22:42:01 +000077 used to open the file a second time, while the named temporary file is
78 still open, varies across platforms (it can be so used on Unix; it cannot
79 on Windows NT or later). If *delete* is true (the default), the file is
80 deleted as soon as it is closed.
Georg Brandl502d9a52009-07-26 15:02:41 +000081 The returned object is always a file-like object whose :attr:`!file`
Christian Heimes81ee3ef2008-05-04 22:42:01 +000082 attribute is the underlying true file object. This file-like object can
83 be used in a :keyword:`with` statement, just like a normal file.
Georg Brandl116aa622007-08-15 14:28:22 +000084
Georg Brandl116aa622007-08-15 14:28:22 +000085
Martin Panter9b566c32015-11-07 00:32:50 +000086.. function:: SpooledTemporaryFile(max_size=0, mode='w+b', buffering=None, encoding=None, newline=None, suffix=None, prefix=None, dir=None)
Georg Brandl116aa622007-08-15 14:28:22 +000087
Christian Heimes81ee3ef2008-05-04 22:42:01 +000088 This function operates exactly as :func:`TemporaryFile` does, except that
89 data is spooled in memory until the file size exceeds *max_size*, or
90 until the file's :func:`fileno` method is called, at which point the
91 contents are written to disk and operation proceeds as with
92 :func:`TemporaryFile`.
Georg Brandl116aa622007-08-15 14:28:22 +000093
Christian Heimes81ee3ef2008-05-04 22:42:01 +000094 The resulting file has one additional method, :func:`rollover`, which
95 causes the file to roll over to an on-disk file regardless of its size.
Georg Brandl116aa622007-08-15 14:28:22 +000096
Christian Heimes81ee3ef2008-05-04 22:42:01 +000097 The returned object is a file-like object whose :attr:`_file` attribute
Martin Panter7462b6492015-11-02 03:37:02 +000098 is either an :class:`io.BytesIO` or :class:`io.StringIO` object (depending on
Serhiy Storchaka4b109cb2013-02-09 11:51:21 +020099 whether binary or text *mode* was specified) or a true file
Serhiy Storchaka4f169a72013-02-09 11:46:42 +0200100 object, depending on whether :func:`rollover` has been called. This
101 file-like object can be used in a :keyword:`with` statement, just like
102 a normal file.
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000103
R David Murrayca76ea12012-10-06 18:32:39 -0400104 .. versionchanged:: 3.3
105 the truncate method now accepts a ``size`` argument.
106
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000107
Martin Panter9b566c32015-11-07 00:32:50 +0000108.. function:: TemporaryDirectory(suffix=None, prefix=None, dir=None)
Nick Coghlan543af752010-10-24 11:23:25 +0000109
Robert Collins2ebdc132015-08-13 11:38:02 +1200110 This function securely creates a temporary directory using the same rules as :func:`mkdtemp`.
Nick Coghlan543af752010-10-24 11:23:25 +0000111 The resulting object can be used as a context manager (see
Robert Collins2ebdc132015-08-13 11:38:02 +1200112 :ref:`tempfile-examples`). On completion of the context or destruction
R David Murray23686072014-02-05 14:53:40 -0500113 of the temporary directory object the newly created temporary directory
Nick Coghlan543af752010-10-24 11:23:25 +0000114 and all its contents are removed from the filesystem.
115
R David Murray23686072014-02-05 14:53:40 -0500116 The directory name can be retrieved from the :attr:`name` attribute of the
117 returned object. When the returned object is used as a context manager, the
118 :attr:`name` will be assigned to the target of the :keyword:`as` clause in
119 the :keyword:`with` statement, if there is one.
Nick Coghlan543af752010-10-24 11:23:25 +0000120
121 The directory can be explicitly cleaned up by calling the
122 :func:`cleanup` method.
123
124 .. versionadded:: 3.2
125
126
Gregory P. Smithad577b92015-05-22 16:18:14 -0700127.. function:: mkstemp(suffix=None, prefix=None, dir=None, text=False)
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000128
129 Creates a temporary file in the most secure manner possible. There are
130 no race conditions in the file's creation, assuming that the platform
131 properly implements the :const:`os.O_EXCL` flag for :func:`os.open`. The
132 file is readable and writable only by the creating user ID. If the
133 platform uses permission bits to indicate whether a file is executable,
134 the file is executable by no one. The file descriptor is not inherited
135 by child processes.
136
137 Unlike :func:`TemporaryFile`, the user of :func:`mkstemp` is responsible
138 for deleting the temporary file when done with it.
139
Martin Panter9b566c32015-11-07 00:32:50 +0000140 If *suffix* is not ``None``, the file name will end with that suffix,
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000141 otherwise there will be no suffix. :func:`mkstemp` does not put a dot
142 between the file name and the suffix; if you need one, put it at the
143 beginning of *suffix*.
144
Martin Panter9b566c32015-11-07 00:32:50 +0000145 If *prefix* is not ``None``, the file name will begin with that prefix;
146 otherwise, a default prefix is used. The default is the return value of
147 :func:`gettempprefix` or :func:`gettempprefixb`, as appropriate.
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000148
Martin Panter9b566c32015-11-07 00:32:50 +0000149 If *dir* is not ``None``, the file will be created in that directory;
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000150 otherwise, a default directory is used. The default directory is chosen
151 from a platform-dependent list, but the user of the application can
152 control the directory location by setting the *TMPDIR*, *TEMP* or *TMP*
153 environment variables. There is thus no guarantee that the generated
154 filename will have any nice properties, such as not requiring quoting
155 when passed to external commands via ``os.popen()``.
156
Martin Panter9b566c32015-11-07 00:32:50 +0000157 If any of *suffix*, *prefix*, and *dir* are not
158 ``None``, they must be the same type.
Gregory P. Smithad577b92015-05-22 16:18:14 -0700159 If they are bytes, the returned name will be bytes instead of str.
160 If you want to force a bytes return value with otherwise default behavior,
161 pass ``suffix=b''``.
162
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000163 If *text* is specified, it indicates whether to open the file in binary
164 mode (the default) or text mode. On some platforms, this makes no
165 difference.
166
167 :func:`mkstemp` returns a tuple containing an OS-level handle to an open
168 file (as would be returned by :func:`os.open`) and the absolute pathname
169 of that file, in that order.
170
Gregory P. Smithad577b92015-05-22 16:18:14 -0700171 .. versionchanged:: 3.5
172 *suffix*, *prefix*, and *dir* may now be supplied in bytes in order to
173 obtain a bytes return value. Prior to this, only str was allowed.
174 *suffix* and *prefix* now accept and default to ``None`` to cause
175 an appropriate default value to be used.
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000176
Gregory P. Smithad577b92015-05-22 16:18:14 -0700177
178.. function:: mkdtemp(suffix=None, prefix=None, dir=None)
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000179
180 Creates a temporary directory in the most secure manner possible. There
181 are no race conditions in the directory's creation. The directory is
182 readable, writable, and searchable only by the creating user ID.
183
184 The user of :func:`mkdtemp` is responsible for deleting the temporary
185 directory and its contents when done with it.
186
187 The *prefix*, *suffix*, and *dir* arguments are the same as for
188 :func:`mkstemp`.
Georg Brandl116aa622007-08-15 14:28:22 +0000189
190 :func:`mkdtemp` returns the absolute pathname of the new directory.
191
Gregory P. Smithad577b92015-05-22 16:18:14 -0700192 .. versionchanged:: 3.5
193 *suffix*, *prefix*, and *dir* may now be supplied in bytes in order to
194 obtain a bytes return value. Prior to this, only str was allowed.
195 *suffix* and *prefix* now accept and default to ``None`` to cause
196 an appropriate default value to be used.
197
Georg Brandl116aa622007-08-15 14:28:22 +0000198
Robert Collins2ebdc132015-08-13 11:38:02 +1200199.. function:: gettempdir()
Georg Brandl116aa622007-08-15 14:28:22 +0000200
Robert Collins2ebdc132015-08-13 11:38:02 +1200201 Return the name of the directory used for temporary files. This
202 defines the default value for the *dir* argument to all functions
203 in this module.
Georg Brandl116aa622007-08-15 14:28:22 +0000204
Robert Collins2ebdc132015-08-13 11:38:02 +1200205 Python searches a standard list of directories to find one which
206 the calling user can create files in. The list is:
Georg Brandl116aa622007-08-15 14:28:22 +0000207
208 #. The directory named by the :envvar:`TMPDIR` environment variable.
209
210 #. The directory named by the :envvar:`TEMP` environment variable.
211
212 #. The directory named by the :envvar:`TMP` environment variable.
213
214 #. A platform-specific location:
215
Georg Brandl116aa622007-08-15 14:28:22 +0000216 * On Windows, the directories :file:`C:\\TEMP`, :file:`C:\\TMP`,
217 :file:`\\TEMP`, and :file:`\\TMP`, in that order.
218
219 * On all other platforms, the directories :file:`/tmp`, :file:`/var/tmp`, and
220 :file:`/usr/tmp`, in that order.
221
222 #. As a last resort, the current working directory.
223
Robert Collins2ebdc132015-08-13 11:38:02 +1200224 The result of this search is cached, see the description of
225 :data:`tempdir` below.
Georg Brandl116aa622007-08-15 14:28:22 +0000226
Gregory P. Smithad577b92015-05-22 16:18:14 -0700227.. function:: gettempdirb()
228
229 Same as :func:`gettempdir` but the return value is in bytes.
230
231 .. versionadded:: 3.5
Georg Brandl116aa622007-08-15 14:28:22 +0000232
Georg Brandl116aa622007-08-15 14:28:22 +0000233.. function:: gettempprefix()
234
235 Return the filename prefix used to create temporary files. This does not
Georg Brandl4b26ff82008-08-04 07:24:52 +0000236 contain the directory component.
Georg Brandl116aa622007-08-15 14:28:22 +0000237
Gregory P. Smithad577b92015-05-22 16:18:14 -0700238.. function:: gettempprefixb()
239
Martin Panter9b566c32015-11-07 00:32:50 +0000240 Same as :func:`gettempprefix` but the return value is in bytes.
Gregory P. Smithad577b92015-05-22 16:18:14 -0700241
242 .. versionadded:: 3.5
243
Robert Collins2ebdc132015-08-13 11:38:02 +1200244The module uses a global variable to store the name of the directory
245used for temporary files returned by :func:`gettempdir`. It can be
246set directly to override the selection process, but this is discouraged.
247All functions in this module take a *dir* argument which can be used
Jelle Zijlstrad3b8f982017-03-11 09:34:55 -0800248to specify the directory and this is the recommended approach.
Robert Collins2ebdc132015-08-13 11:38:02 +1200249
250.. data:: tempdir
251
252 When set to a value other than ``None``, this variable defines the
Jelle Zijlstrad3b8f982017-03-11 09:34:55 -0800253 default value for the *dir* argument to the functions defined in this
Robert Collins2ebdc132015-08-13 11:38:02 +1200254 module.
255
256 If ``tempdir`` is unset or ``None`` at any call to any of the above
Martin Panter8f265652016-04-19 04:03:41 +0000257 functions except :func:`gettempprefix` it is initialized following the
Robert Collins2ebdc132015-08-13 11:38:02 +1200258 algorithm described in :func:`gettempdir`.
259
260.. _tempfile-examples:
Nick Coghlan543af752010-10-24 11:23:25 +0000261
262Examples
263--------
264
265Here are some examples of typical usage of the :mod:`tempfile` module::
266
267 >>> import tempfile
268
269 # create a temporary file and write some data to it
270 >>> fp = tempfile.TemporaryFile()
Ross Lagerwall810b94a2011-04-10 09:30:04 +0200271 >>> fp.write(b'Hello world!')
Nick Coghlan543af752010-10-24 11:23:25 +0000272 # read data from file
273 >>> fp.seek(0)
274 >>> fp.read()
Ross Lagerwall810b94a2011-04-10 09:30:04 +0200275 b'Hello world!'
Nick Coghlan543af752010-10-24 11:23:25 +0000276 # close the file, it will be removed
277 >>> fp.close()
278
279 # create a temporary file using a context manager
280 >>> with tempfile.TemporaryFile() as fp:
Ross Lagerwall810b94a2011-04-10 09:30:04 +0200281 ... fp.write(b'Hello world!')
Nick Coghlan543af752010-10-24 11:23:25 +0000282 ... fp.seek(0)
283 ... fp.read()
Ross Lagerwall810b94a2011-04-10 09:30:04 +0200284 b'Hello world!'
Nick Coghlan543af752010-10-24 11:23:25 +0000285 >>>
286 # file is now closed and removed
287
288 # create a temporary directory using the context manager
289 >>> with tempfile.TemporaryDirectory() as tmpdirname:
Ross Lagerwall810b94a2011-04-10 09:30:04 +0200290 ... print('created temporary directory', tmpdirname)
Nick Coghlan543af752010-10-24 11:23:25 +0000291 >>>
292 # directory and contents have been removed
293
Robert Collins2ebdc132015-08-13 11:38:02 +1200294
295Deprecated functions and variables
296----------------------------------
297
298A historical way to create temporary files was to first generate a
299file name with the :func:`mktemp` function and then create a file
300using this name. Unfortunately this is not secure, because a different
301process may create a file with this name in the time between the call
302to :func:`mktemp` and the subsequent attempt to create the file by the
303first process. The solution is to combine the two steps and create the
304file immediately. This approach is used by :func:`mkstemp` and the
305other functions described above.
306
307.. function:: mktemp(suffix='', prefix='tmp', dir=None)
308
309 .. deprecated:: 2.3
310 Use :func:`mkstemp` instead.
311
312 Return an absolute pathname of a file that did not exist at the time the
Martin Panter9b566c32015-11-07 00:32:50 +0000313 call is made. The *prefix*, *suffix*, and *dir* arguments are similar
314 to those of :func:`mkstemp`, except that bytes file names, ``suffix=None``
315 and ``prefix=None`` are not supported.
Robert Collins2ebdc132015-08-13 11:38:02 +1200316
317 .. warning::
318
319 Use of this function may introduce a security hole in your program. By
320 the time you get around to doing anything with the file name it returns,
321 someone else may have beaten you to the punch. :func:`mktemp` usage can
322 be replaced easily with :func:`NamedTemporaryFile`, passing it the
323 ``delete=False`` parameter::
324
325 >>> f = NamedTemporaryFile(delete=False)
326 >>> f.name
327 '/tmp/tmptjujjt'
328 >>> f.write(b"Hello World!\n")
329 13
330 >>> f.close()
331 >>> os.unlink(f.name)
332 >>> os.path.exists(f.name)
333 False