import os

from django.core.files.uploadedfile import UploadedFile

from django.http import HttpResponse, HttpResponseServerError

from django.utils import simplejson

from models import FileModel, UPLOAD_TO

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)

    full_name = u'%s/%s' % (UPLOAD_TO, uni_named_file.name)

    if not os.path.exists(full_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)