|
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374 |
- """
- The temp module provides a NamedTemporaryFile that can be reopened in the same
- process on any platform. Most platforms use the standard Python
- tempfile.NamedTemporaryFile class, but Windows users are given a custom class.
-
- This is needed because the Python implementation of NamedTemporaryFile uses the
- O_TEMPORARY flag under Windows, which prevents the file from being reopened
- if the same flag is not provided [1][2]. Note that this does not address the
- more general issue of opening a file for writing and reading in multiple
- processes in a manner that works across platforms.
-
- The custom version of NamedTemporaryFile doesn't support the same keyword
- arguments available in tempfile.NamedTemporaryFile.
-
- 1: https://mail.python.org/pipermail/python-list/2005-December/336957.html
- 2: https://bugs.python.org/issue14243
- """
-
- import os
- import tempfile
-
- from django.core.files.utils import FileProxyMixin
-
- __all__ = ('NamedTemporaryFile', 'gettempdir',)
-
-
- if os.name == 'nt':
- class TemporaryFile(FileProxyMixin):
- """
- Temporary file object constructor that supports reopening of the
- temporary file in Windows.
-
- Unlike tempfile.NamedTemporaryFile from the standard library,
- __init__() doesn't support the 'delete', 'buffering', 'encoding', or
- 'newline' keyword arguments.
- """
- def __init__(self, mode='w+b', bufsize=-1, suffix='', prefix='', dir=None):
- fd, name = tempfile.mkstemp(suffix=suffix, prefix=prefix, dir=dir)
- self.name = name
- self.file = os.fdopen(fd, mode, bufsize)
- self.close_called = False
-
- # Because close can be called during shutdown
- # we need to cache os.unlink and access it
- # as self.unlink only
- unlink = os.unlink
-
- def close(self):
- if not self.close_called:
- self.close_called = True
- try:
- self.file.close()
- except (OSError, IOError):
- pass
- try:
- self.unlink(self.name)
- except OSError:
- pass
-
- def __del__(self):
- self.close()
-
- def __enter__(self):
- self.file.__enter__()
- return self
-
- def __exit__(self, exc, value, tb):
- self.file.__exit__(exc, value, tb)
-
- NamedTemporaryFile = TemporaryFile
- else:
- NamedTemporaryFile = tempfile.NamedTemporaryFile
-
- gettempdir = tempfile.gettempdir
|