mediacms/uploader/views.py
2020-12-15 23:33:43 +02:00

77 lines
2.7 KiB
Python

# -*- coding: utf-8 -*-
import os
import shutil
from django.http import JsonResponse
from django.views import generic
from django.conf import settings
from django.core.files import File
from django.core.exceptions import PermissionDenied
from cms.permissions import user_allowed_to_upload
from files.models import Media
from files.helpers import rm_file
from .forms import FineUploaderUploadForm, FineUploaderUploadSuccessForm
from .fineuploader import ChunkedFineUploader
class FineUploaderView(generic.FormView):
http_method_names = ("post",)
form_class_upload = FineUploaderUploadForm
form_class_upload_success = FineUploaderUploadSuccessForm
@property
def concurrent(self):
return settings.CONCURRENT_UPLOADS
@property
def chunks_done(self):
return self.chunks_done_param_name in self.request.GET
@property
def chunks_done_param_name(self):
return settings.CHUNKS_DONE_PARAM_NAME
def make_response(self, data, **kwargs):
return JsonResponse(data, **kwargs)
def get_form(self, form_class=None):
if self.chunks_done:
form_class = self.form_class_upload_success
else:
form_class = self.form_class_upload
return form_class(**self.get_form_kwargs())
def dispatch(self, request, *args, **kwargs):
if not user_allowed_to_upload(request):
raise PermissionDenied # HTTP 403
return super(FineUploaderView, self).dispatch(request, *args, **kwargs)
def form_valid(self, form):
self.upload = ChunkedFineUploader(form.cleaned_data, self.concurrent)
if self.upload.concurrent and self.chunks_done:
try:
self.upload.combine_chunks()
except FileNotFoundError:
data = {"success": False, "error": "Error with File Uploading"}
return self.make_response(data, status=400)
elif self.upload.total_parts == 1:
self.upload.save()
else:
self.upload.save()
return self.make_response({"success": True})
# create media!
media_file = os.path.join(settings.MEDIA_ROOT, self.upload.real_path)
with open(media_file, "rb") as f:
myfile = File(f)
new = Media.objects.create(media_file=myfile, user=self.request.user)
rm_file(media_file)
shutil.rmtree(os.path.join(settings.MEDIA_ROOT, self.upload.file_path))
return self.make_response(
{"success": True, "media_url": new.get_absolute_url()}
)
def form_invalid(self, form):
data = {"success": False, "error": "%s" % repr(form.errors)}
return self.make_response(data, status=400)