from django.core.files.uploadhandler import FileUploadHandler
import time


class DebugFileUploader(FileUploadHandler):
    def __init__(self, request=None):
        super(DebugFileUploader, self).__init__(request)
        self.count = 0

    def receive_data_chunk(self, raw_data, start):
        time.sleep(1)
        self.count = self.count + len(raw_data)
        fail_at = None
        if 'fail_at' in self.request.GET:
            fail_at = int(self.request.GET.get('fail_at'))
        if fail_at and self.count > fail_at:
            raise Exception('Triggered fail')

        return raw_data

    def file_complete(self, file_size):
        return None