Fixed #7658 -- Added some Windows-specific tempfile handling. The standard
stuff doesn't work with the way Django's file uploading code wants to operate. Patch from Mike Axiak. git-svn-id: http://code.djangoproject.com/svn/django/trunk@8096 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
parent
e29aece743
commit
103d484807
|
@ -0,0 +1,58 @@
|
||||||
|
"""
|
||||||
|
The temp module provides a NamedTemporaryFile that can be re-opened on any
|
||||||
|
platform. Most platforms use the standard Python tempfile.TemporaryFile class,
|
||||||
|
but MS Windows users are given a custom class.
|
||||||
|
|
||||||
|
This is needed because in Windows NT, the default implementation of
|
||||||
|
NamedTemporaryFile uses the O_TEMPORARY flag, and thus cannot be reopened [1].
|
||||||
|
|
||||||
|
1: http://mail.python.org/pipermail/python-list/2005-December/359474.html
|
||||||
|
"""
|
||||||
|
|
||||||
|
import os
|
||||||
|
import tempfile
|
||||||
|
|
||||||
|
__all__ = ('NamedTemporaryFile', 'gettempdir',)
|
||||||
|
|
||||||
|
if os.name == 'nt':
|
||||||
|
class TemporaryFile(object):
|
||||||
|
"""
|
||||||
|
Temporary file object constructor that works in Windows and supports
|
||||||
|
reopening of the temporary file in windows.
|
||||||
|
"""
|
||||||
|
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)
|
||||||
|
|
||||||
|
def __del__(self):
|
||||||
|
try:
|
||||||
|
self._file.close()
|
||||||
|
except (OSError, IOError):
|
||||||
|
pass
|
||||||
|
try:
|
||||||
|
os.unlink(self.name)
|
||||||
|
except (OSError):
|
||||||
|
pass
|
||||||
|
|
||||||
|
try:
|
||||||
|
super(TemporaryFile, self).__del__()
|
||||||
|
except AttributeError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
def read(self, *args): return self._file.read(*args)
|
||||||
|
def seek(self, offset): return self._file.seek(offset)
|
||||||
|
def write(self, s): return self._file.write(s)
|
||||||
|
def close(self): return self._file.close()
|
||||||
|
def __iter__(self): return iter(self._file)
|
||||||
|
def readlines(self, size=None): return self._file.readlines(size)
|
||||||
|
def xreadlines(self): return self._file.xreadlines()
|
||||||
|
|
||||||
|
NamedTemporaryFile = TemporaryFile
|
||||||
|
else:
|
||||||
|
NamedTemporaryFile = tempfile.NamedTemporaryFile
|
||||||
|
|
||||||
|
gettempdir = tempfile.gettempdir
|
|
@ -3,7 +3,6 @@ Classes representing uploaded files.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
import os
|
import os
|
||||||
import tempfile
|
|
||||||
import warnings
|
import warnings
|
||||||
try:
|
try:
|
||||||
from cStringIO import StringIO
|
from cStringIO import StringIO
|
||||||
|
@ -12,6 +11,8 @@ except ImportError:
|
||||||
|
|
||||||
from django.conf import settings
|
from django.conf import settings
|
||||||
|
|
||||||
|
from django.core.files import temp as tempfile
|
||||||
|
|
||||||
__all__ = ('UploadedFile', 'TemporaryUploadedFile', 'InMemoryUploadedFile', 'SimpleUploadedFile')
|
__all__ = ('UploadedFile', 'TemporaryUploadedFile', 'InMemoryUploadedFile', 'SimpleUploadedFile')
|
||||||
|
|
||||||
# Because we fooled around with it a bunch, UploadedFile has a bunch
|
# Because we fooled around with it a bunch, UploadedFile has a bunch
|
||||||
|
|
|
@ -2,9 +2,9 @@ import os
|
||||||
import errno
|
import errno
|
||||||
import sha
|
import sha
|
||||||
import shutil
|
import shutil
|
||||||
import tempfile
|
|
||||||
import unittest
|
import unittest
|
||||||
|
|
||||||
|
from django.core.files import temp as tempfile
|
||||||
from django.core.files.uploadedfile import SimpleUploadedFile
|
from django.core.files.uploadedfile import SimpleUploadedFile
|
||||||
from django.test import TestCase, client
|
from django.test import TestCase, client
|
||||||
from django.utils import simplejson
|
from django.utils import simplejson
|
||||||
|
|
Loading…
Reference in New Issue