/tests/regressiontests/file_uploads/views.py

  1import os
  2from django.core.files.uploadedfile import UploadedFile
  3from django.http import HttpResponse, HttpResponseServerError
  4from django.utils import simplejson
  5from models import FileModel, UPLOAD_TO
  6from uploadhandler import QuotaUploadHandler, ErroringUploadHandler
  7from django.utils.hashcompat import sha_constructor
  8from tests import UNICODE_FILENAME
  9
 10def file_upload_view(request):
 11    """
 12    Check that a file upload can be updated into the POST dictionary without
 13    going pear-shaped.
 14    """
 15    form_data = request.POST.copy()
 16    form_data.update(request.FILES)
 17    if isinstance(form_data.get('file_field'), UploadedFile) and isinstance(form_data['name'], unicode):
 18        # If a file is posted, the dummy client should only post the file name,
 19        # not the full path.
 20        if os.path.dirname(form_data['file_field'].name) != '':
 21            return HttpResponseServerError()
 22        return HttpResponse('')
 23    else:
 24        return HttpResponseServerError()
 25
 26def file_upload_view_verify(request):
 27    """
 28    Use the sha digest hash to verify the uploaded contents.
 29    """
 30    form_data = request.POST.copy()
 31    form_data.update(request.FILES)
 32
 33    for key, value in form_data.items():
 34        if key.endswith('_hash'):
 35            continue
 36        if key + '_hash' not in form_data:
 37            continue
 38        submitted_hash = form_data[key + '_hash']
 39        if isinstance(value, UploadedFile):
 40            new_hash = sha_constructor(value.read()).hexdigest()
 41        else:
 42            new_hash = sha_constructor(value).hexdigest()
 43        if new_hash != submitted_hash:
 44            return HttpResponseServerError()
 45
 46    # Adding large file to the database should succeed
 47    largefile = request.FILES['file_field2']
 48    obj = FileModel()
 49    obj.testfile.save(largefile.name, largefile)
 50
 51    return HttpResponse('')
 52
 53def file_upload_unicode_name(request):
 54
 55    # Check to see if unicode name came through properly.
 56    if not request.FILES['file_unicode'].name.endswith(UNICODE_FILENAME):
 57        return HttpResponseServerError()
 58
 59    response = None
 60
 61    # Check to make sure the exotic characters are preserved even
 62    # through file save.
 63    uni_named_file = request.FILES['file_unicode']
 64    obj = FileModel.objects.create(testfile=uni_named_file)
 65    full_name = u'%s/%s' % (UPLOAD_TO, uni_named_file.name)
 66    if not os.path.exists(full_name):
 67        response = HttpResponseServerError()
 68
 69    # Cleanup the object with its exotic file name immediately.
 70    # (shutil.rmtree used elsewhere in the tests to clean up the
 71    # upload directory has been seen to choke on unicode
 72    # filenames on Windows.)
 73    obj.delete()
 74    os.unlink(full_name)
 75
 76    if response:
 77        return response
 78    else:
 79        return HttpResponse('')
 80
 81def file_upload_echo(request):
 82    """
 83    Simple view to echo back info about uploaded files for tests.
 84    """
 85    r = dict([(k, f.name) for k, f in request.FILES.items()])
 86    return HttpResponse(simplejson.dumps(r))
 87
 88def file_upload_quota(request):
 89    """
 90    Dynamically add in an upload handler.
 91    """
 92    request.upload_handlers.insert(0, QuotaUploadHandler())
 93    return file_upload_echo(request)
 94
 95def file_upload_quota_broken(request):
 96    """
 97    You can't change handlers after reading FILES; this view shouldn't work.
 98    """
 99    response = file_upload_echo(request)
100    request.upload_handlers.insert(0, QuotaUploadHandler())
101    return response
102
103def file_upload_getlist_count(request):
104    """
105    Check the .getlist() function to ensure we receive the correct number of files.
106    """
107    file_counts = {}
108
109    for key in request.FILES.keys():
110        file_counts[key] = len(request.FILES.getlist(key))
111    return HttpResponse(simplejson.dumps(file_counts))
112
113def file_upload_errors(request):
114    request.upload_handlers.insert(0, ErroringUploadHandler())
115    return file_upload_echo(request)