114 lines
3.7 KiB
Python
114 lines
3.7 KiB
Python
import os
|
|
from django.core.files.uploadedfile import UploadedFile
|
|
from django.http import HttpResponse, HttpResponseServerError
|
|
from django.utils import simplejson
|
|
from models import FileModel
|
|
from uploadhandler import QuotaUploadHandler, ErroringUploadHandler
|
|
from django.utils.hashcompat import sha_constructor
|
|
from tests import UNICODE_FILENAME
|
|
|
|
def file_upload_view(request):
|
|
"""
|
|
Check that a file upload can be updated into the POST dictionary without
|
|
going pear-shaped.
|
|
"""
|
|
form_data = request.POST.copy()
|
|
form_data.update(request.FILES)
|
|
if isinstance(form_data.get('file_field'), UploadedFile) and isinstance(form_data['name'], unicode):
|
|
# If a file is posted, the dummy client should only post the file name,
|
|
# not the full path.
|
|
if os.path.dirname(form_data['file_field'].name) != '':
|
|
return HttpResponseServerError()
|
|
return HttpResponse('')
|
|
else:
|
|
return HttpResponseServerError()
|
|
|
|
def file_upload_view_verify(request):
|
|
"""
|
|
Use the sha digest hash to verify the uploaded contents.
|
|
"""
|
|
form_data = request.POST.copy()
|
|
form_data.update(request.FILES)
|
|
|
|
for key, value in form_data.items():
|
|
if key.endswith('_hash'):
|
|
continue
|
|
if key + '_hash' not in form_data:
|
|
continue
|
|
submitted_hash = form_data[key + '_hash']
|
|
if isinstance(value, UploadedFile):
|
|
new_hash = sha_constructor(value.read()).hexdigest()
|
|
else:
|
|
new_hash = sha_constructor(value).hexdigest()
|
|
if new_hash != submitted_hash:
|
|
return HttpResponseServerError()
|
|
|
|
# Adding large file to the database should succeed
|
|
largefile = request.FILES['file_field2']
|
|
obj = FileModel()
|
|
obj.testfile.save(largefile.name, largefile)
|
|
|
|
return HttpResponse('')
|
|
|
|
def file_upload_unicode_name(request):
|
|
|
|
# Check to see if unicode name came through properly.
|
|
if not request.FILES['file_unicode'].name.endswith(UNICODE_FILENAME):
|
|
return HttpResponseServerError()
|
|
|
|
response = None
|
|
|
|
# Check to make sure the exotic characters are preserved even
|
|
# through file save.
|
|
uni_named_file = request.FILES['file_unicode']
|
|
obj = FileModel.objects.create(testfile=uni_named_file)
|
|
if not obj.testfile.name.endswith(uni_named_file.name):
|
|
response = HttpResponseServerError()
|
|
|
|
# Cleanup the object with its exotic file name immediately.
|
|
# (shutil.rmtree used elsewhere in the tests to clean up the
|
|
# upload directory has been seen to choke on unicode
|
|
# filenames on Windows.)
|
|
obj.delete()
|
|
|
|
if response:
|
|
return response
|
|
else:
|
|
return HttpResponse('')
|
|
|
|
def file_upload_echo(request):
|
|
"""
|
|
Simple view to echo back info about uploaded files for tests.
|
|
"""
|
|
r = dict([(k, f.name) for k, f in request.FILES.items()])
|
|
return HttpResponse(simplejson.dumps(r))
|
|
|
|
def file_upload_quota(request):
|
|
"""
|
|
Dynamically add in an upload handler.
|
|
"""
|
|
request.upload_handlers.insert(0, QuotaUploadHandler())
|
|
return file_upload_echo(request)
|
|
|
|
def file_upload_quota_broken(request):
|
|
"""
|
|
You can't change handlers after reading FILES; this view shouldn't work.
|
|
"""
|
|
response = file_upload_echo(request)
|
|
request.upload_handlers.insert(0, QuotaUploadHandler())
|
|
return response
|
|
|
|
def file_upload_getlist_count(request):
|
|
"""
|
|
Check the .getlist() function to ensure we receive the correct number of files.
|
|
"""
|
|
file_counts = {}
|
|
|
|
for key in request.FILES.keys():
|
|
file_counts[key] = len(request.FILES.getlist(key))
|
|
return HttpResponse(simplejson.dumps(file_counts))
|
|
|
|
def file_upload_errors(request):
|
|
request.upload_handlers.insert(0, ErroringUploadHandler())
|
|
return file_upload_echo(request)
|