tempfile.rst 11 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14
:mod:`tempfile` --- Generate temporary files and directories
============================================================

.. sectionauthor:: Zack Weinberg <zack@codesourcery.com>


.. module:: tempfile
   :synopsis: Generate temporary files and directories.


.. index::
   pair: temporary; file name
   pair: temporary; file

Raymond Hettinger's avatar
Raymond Hettinger committed
15 16 17 18
**Source code:** :source:`Lib/tempfile.py`

--------------

19
This module generates temporary files and directories.  It works on all
20 21 22 23 24
supported platforms.  It provides three new functions,
:func:`NamedTemporaryFile`, :func:`mkstemp`, and :func:`mkdtemp`, which should
eliminate all remaining need to use the insecure :func:`mktemp` function.
Temporary file names created by this module no longer contain the process ID;
instead a string of six random characters is used.
25

Christian Heimes's avatar
Christian Heimes committed
26 27
Also, all the user-callable functions now take additional arguments which
allow direct control over the location and name of temporary files.  It is
28
no longer necessary to use the global *tempdir* variable.
Christian Heimes's avatar
Christian Heimes committed
29 30
To maintain backward compatibility, the argument order is somewhat odd; it
is recommended to use keyword arguments for clarity.
31

32
The module defines the following user-callable items:
33

34
.. function:: TemporaryFile(mode='w+b', buffering=None, encoding=None, newline=None, suffix='', prefix='tmp', dir=None)
35

36
   Return a :term:`file-like object` that can be used as a temporary storage area.
Christian Heimes's avatar
Christian Heimes committed
37
   The file is created using :func:`mkstemp`. It will be destroyed as soon
38
   as it is closed (including an implicit close when the object is garbage
Christian Heimes's avatar
Christian Heimes committed
39 40 41 42
   collected).  Under Unix, the directory entry for the file is removed
   immediately after the file is created.  Other platforms do not support
   this; your code should not rely on a temporary file created using this
   function having or not having a visible name in the file system.
43

Christian Heimes's avatar
Christian Heimes committed
44 45 46
   The *mode* parameter defaults to ``'w+b'`` so that the file created can
   be read and written without being closed.  Binary mode is used so that it
   behaves consistently on all platforms without regard for the data that is
47 48
   stored.  *buffering*, *encoding* and *newline* are interpreted as for
   :func:`open`.
49 50 51

   The *dir*, *prefix* and *suffix* parameters are passed to :func:`mkstemp`.

52
   The returned object is a true file object on POSIX platforms.  On other
53
   platforms, it is a file-like object whose :attr:`!file` attribute is the
Christian Heimes's avatar
Christian Heimes committed
54 55
   underlying true file object. This file-like object can be used in a
   :keyword:`with` statement, just like a normal file.
56

57

58
.. function:: NamedTemporaryFile(mode='w+b', buffering=None, encoding=None, newline=None, suffix='', prefix='tmp', dir=None, delete=True)
59

Christian Heimes's avatar
Christian Heimes committed
60 61 62
   This function operates exactly as :func:`TemporaryFile` does, except that
   the file is guaranteed to have a visible name in the file system (on
   Unix, the directory entry is not unlinked).  That name can be retrieved
63
   from the :attr:`name` attribute of the file object.  Whether the name can be
Christian Heimes's avatar
Christian Heimes committed
64 65 66 67
   used to open the file a second time, while the named temporary file is
   still open, varies across platforms (it can be so used on Unix; it cannot
   on Windows NT or later).  If *delete* is true (the default), the file is
   deleted as soon as it is closed.
68
   The returned object is always a file-like object whose :attr:`!file`
Christian Heimes's avatar
Christian Heimes committed
69 70
   attribute is the underlying true file object. This file-like object can
   be used in a :keyword:`with` statement, just like a normal file.
71 72


73
.. function:: SpooledTemporaryFile(max_size=0, mode='w+b', buffering=None, encoding=None, newline=None, suffix='', prefix='tmp', dir=None)
74

Christian Heimes's avatar
Christian Heimes committed
75 76 77 78 79
   This function operates exactly as :func:`TemporaryFile` does, except that
   data is spooled in memory until the file size exceeds *max_size*, or
   until the file's :func:`fileno` method is called, at which point the
   contents are written to disk and operation proceeds as with
   :func:`TemporaryFile`.
80

Christian Heimes's avatar
Christian Heimes committed
81 82
   The resulting file has one additional method, :func:`rollover`, which
   causes the file to roll over to an on-disk file regardless of its size.
83

84
   The returned object is a file-like object whose :attr:`_file` attribute
85
   is either a :class:`io.BytesIO` or :class:`io.StringIO` object (depending on
86
   whether binary or text *mode* was specified) or a true file
87 88 89
   object, depending on whether :func:`rollover` has been called.  This
   file-like object can be used in a :keyword:`with` statement, just like
   a normal file.
90

91 92 93
   .. versionchanged:: 3.3
      the truncate method now accepts a ``size`` argument.

94

95 96 97 98 99
.. function:: TemporaryDirectory(suffix='', prefix='tmp', dir=None)

   This function creates a temporary directory using :func:`mkdtemp`
   (the supplied arguments are passed directly to the underlying function).
   The resulting object can be used as a context manager (see
100 101
   :ref:`context-managers`).  On completion of the context or destruction
   of the temporary directory object the newly created temporary directory
102 103
   and all its contents are removed from the filesystem.

104 105 106 107
   The directory name can be retrieved from the :attr:`name` attribute of the
   returned object.  When the returned object is used as a context manager, the
   :attr:`name` will be assigned to the target of the :keyword:`as` clause in
   the :keyword:`with` statement, if there is one.
108 109 110 111 112 113 114

   The directory can be explicitly cleaned up by calling the
   :func:`cleanup` method.

   .. versionadded:: 3.2


115
.. function:: mkstemp(suffix='', prefix='tmp', dir=None, text=False)
116

Christian Heimes's avatar
Christian Heimes committed
117 118 119 120 121 122 123
   Creates a temporary file in the most secure manner possible.  There are
   no race conditions in the file's creation, assuming that the platform
   properly implements the :const:`os.O_EXCL` flag for :func:`os.open`.  The
   file is readable and writable only by the creating user ID.  If the
   platform uses permission bits to indicate whether a file is executable,
   the file is executable by no one.  The file descriptor is not inherited
   by child processes.
124

Christian Heimes's avatar
Christian Heimes committed
125 126
   Unlike :func:`TemporaryFile`, the user of :func:`mkstemp` is responsible
   for deleting the temporary file when done with it.
127

Christian Heimes's avatar
Christian Heimes committed
128 129 130 131
   If *suffix* is specified, the file name will end with that suffix,
   otherwise there will be no suffix.  :func:`mkstemp` does not put a dot
   between the file name and the suffix; if you need one, put it at the
   beginning of *suffix*.
132

Christian Heimes's avatar
Christian Heimes committed
133 134
   If *prefix* is specified, the file name will begin with that prefix;
   otherwise, a default prefix is used.
135

Christian Heimes's avatar
Christian Heimes committed
136 137 138 139 140 141 142
   If *dir* is specified, the file will be created in that directory;
   otherwise, a default directory is used.  The default directory is chosen
   from a platform-dependent list, but the user of the application can
   control the directory location by setting the *TMPDIR*, *TEMP* or *TMP*
   environment variables.  There is thus no guarantee that the generated
   filename will have any nice properties, such as not requiring quoting
   when passed to external commands via ``os.popen()``.
143

Christian Heimes's avatar
Christian Heimes committed
144 145 146
   If *text* is specified, it indicates whether to open the file in binary
   mode (the default) or text mode.  On some platforms, this makes no
   difference.
147

Christian Heimes's avatar
Christian Heimes committed
148 149 150
   :func:`mkstemp` returns a tuple containing an OS-level handle to an open
   file (as would be returned by :func:`os.open`) and the absolute pathname
   of that file, in that order.
151 152


153
.. function:: mkdtemp(suffix='', prefix='tmp', dir=None)
154

Christian Heimes's avatar
Christian Heimes committed
155 156 157
   Creates a temporary directory in the most secure manner possible. There
   are no race conditions in the directory's creation.  The directory is
   readable, writable, and searchable only by the creating user ID.
158

Christian Heimes's avatar
Christian Heimes committed
159 160
   The user of :func:`mkdtemp` is responsible for deleting the temporary
   directory and its contents when done with it.
161

Christian Heimes's avatar
Christian Heimes committed
162 163
   The *prefix*, *suffix*, and *dir* arguments are the same as for
   :func:`mkstemp`.
164 165 166 167

   :func:`mkdtemp` returns the absolute pathname of the new directory.


168
.. function:: mktemp(suffix='', prefix='tmp', dir=None)
169 170 171 172

   .. deprecated:: 2.3
      Use :func:`mkstemp` instead.

Christian Heimes's avatar
Christian Heimes committed
173 174 175
   Return an absolute pathname of a file that did not exist at the time the
   call is made.  The *prefix*, *suffix*, and *dir* arguments are the same
   as for :func:`mkstemp`.
176 177 178

   .. warning::

179 180 181 182 183
      Use of this function may introduce a security hole in your program.  By
      the time you get around to doing anything with the file name it returns,
      someone else may have beaten you to the punch.  :func:`mktemp` usage can
      be replaced easily with :func:`NamedTemporaryFile`, passing it the
      ``delete=False`` parameter::
184 185 186

         >>> f = NamedTemporaryFile(delete=False)
         >>> f.name
187 188 189
         '/tmp/tmptjujjt'
         >>> f.write(b"Hello World!\n")
         13
190 191 192 193
         >>> f.close()
         >>> os.unlink(f.name)
         >>> os.path.exists(f.name)
         False
194

Christian Heimes's avatar
Christian Heimes committed
195 196 197 198
The module uses two global variables that tell it how to construct a
temporary name.  They are initialized at the first call to any of the
functions above.  The caller may change them, but this is discouraged; use
the appropriate function arguments, instead.
199 200 201 202


.. data:: tempdir

Christian Heimes's avatar
Christian Heimes committed
203 204 205
   When set to a value other than ``None``, this variable defines the
   default value for the *dir* argument to all the functions defined in this
   module.
206

Christian Heimes's avatar
Christian Heimes committed
207 208 209 210
   If ``tempdir`` is unset or ``None`` at any call to any of the above
   functions, Python searches a standard list of directories and sets
   *tempdir* to the first one which the calling user can create files in.
   The list is:
211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238

   #. The directory named by the :envvar:`TMPDIR` environment variable.

   #. The directory named by the :envvar:`TEMP` environment variable.

   #. The directory named by the :envvar:`TMP` environment variable.

   #. A platform-specific location:

      * On Windows, the directories :file:`C:\\TEMP`, :file:`C:\\TMP`,
        :file:`\\TEMP`, and :file:`\\TMP`, in that order.

      * On all other platforms, the directories :file:`/tmp`, :file:`/var/tmp`, and
        :file:`/usr/tmp`, in that order.

   #. As a last resort, the current working directory.


.. function:: gettempdir()

   Return the directory currently selected to create temporary files in. If
   :data:`tempdir` is not ``None``, this simply returns its contents; otherwise,
   the search described above is performed, and the result returned.


.. function:: gettempprefix()

   Return the filename prefix used to create temporary files.  This does not
239
   contain the directory component.
240

241 242 243 244 245 246 247 248 249 250

Examples
--------

Here are some examples of typical usage of the :mod:`tempfile` module::

    >>> import tempfile

    # create a temporary file and write some data to it
    >>> fp = tempfile.TemporaryFile()
251
    >>> fp.write(b'Hello world!')
252 253 254
    # read data from file
    >>> fp.seek(0)
    >>> fp.read()
255
    b'Hello world!'
256 257 258 259 260
    # close the file, it will be removed
    >>> fp.close()

    # create a temporary file using a context manager
    >>> with tempfile.TemporaryFile() as fp:
261
    ...     fp.write(b'Hello world!')
262 263
    ...     fp.seek(0)
    ...     fp.read()
264
    b'Hello world!'
265 266 267 268 269
    >>>
    # file is now closed and removed

    # create a temporary directory using the context manager
    >>> with tempfile.TemporaryDirectory() as tmpdirname:
270
    ...     print('created temporary directory', tmpdirname)
271 272 273
    >>>
    # directory and contents have been removed