94 lines
3.2 KiB
Python
Executable file
94 lines
3.2 KiB
Python
Executable file
# https://github.com/FineUploader/server-examples/blob/master/python/django-fine-uploader
|
|
|
|
from django.conf import settings
|
|
from django import forms
|
|
from django.forms import ClearableFileInput
|
|
|
|
import os, os.path
|
|
import tempfile
|
|
import shutil
|
|
|
|
__all__ = ("handle_upload", "save_upload", "FineUploadForm", "FineUploadFileInput")
|
|
|
|
|
|
def combine_chunks(total_parts, total_size, source_folder, dest):
|
|
if not os.path.exists(os.path.dirname(dest)):
|
|
os.makedirs(os.path.dirname(dest))
|
|
|
|
with open(dest, "wb+") as destination:
|
|
for i in range(total_parts):
|
|
part = os.path.join(source_folder, str(i))
|
|
with open(part, "rb") as source:
|
|
destination.write(source.read())
|
|
|
|
|
|
def save_upload(f, path):
|
|
if not os.path.exists(os.path.dirname(path)):
|
|
os.makedirs(os.path.dirname(path))
|
|
with open(path, "wb+") as destination:
|
|
if hasattr(f, "multiple_chunks") and f.multiple_chunks():
|
|
for chunk in f.chunks():
|
|
destination.write(chunk)
|
|
else:
|
|
destination.write(f.read())
|
|
|
|
|
|
# pass callback function to post_upload
|
|
def handle_upload(f, fileattrs, upload_dir, post_upload=None):
|
|
chunks_dir = os.path.join(tempfile.gettempdir(), "chunk_upload_tmp")
|
|
if not os.path.exists(os.path.dirname(chunks_dir)):
|
|
os.makedirs(os.path.dirname(chunks_dir))
|
|
chunked = False
|
|
dest_folder = upload_dir
|
|
dest = os.path.join(dest_folder, fileattrs["qqfilename"])
|
|
|
|
# Chunked
|
|
if fileattrs.get("qqtotalparts") and int(fileattrs["qqtotalparts"]) > 1:
|
|
chunked = True
|
|
dest_folder = os.path.join(chunks_dir, fileattrs["qquuid"])
|
|
dest = os.path.join(
|
|
dest_folder, fileattrs["qqfilename"], str(fileattrs["qqpartindex"])
|
|
)
|
|
save_upload(f, dest)
|
|
|
|
# If the last chunk has been sent, combine the parts.
|
|
if chunked and (fileattrs["qqtotalparts"] - 1 == fileattrs["qqpartindex"]):
|
|
combine_chunks(
|
|
fileattrs["qqtotalparts"],
|
|
fileattrs["qqtotalfilesize"],
|
|
source_folder=os.path.dirname(dest),
|
|
dest=os.path.join(upload_dir, fileattrs["qqfilename"]),
|
|
)
|
|
shutil.rmtree(os.path.dirname(os.path.dirname(dest)))
|
|
|
|
if post_upload and (
|
|
not chunked or fileattrs["qqtotalparts"] - 1 == fileattrs["qqpartindex"]
|
|
):
|
|
post_upload()
|
|
|
|
|
|
class FineUploadForm(forms.Form):
|
|
qqfile = forms.FileField()
|
|
qquuid = forms.CharField()
|
|
qqfilename = forms.CharField()
|
|
qqpartindex = forms.IntegerField(required=False)
|
|
qqchunksize = forms.IntegerField(required=False)
|
|
qqpartbyteoffset = forms.IntegerField(required=False)
|
|
qqtotalfilesize = forms.IntegerField(required=False)
|
|
qqtotalparts = forms.IntegerField(required=False)
|
|
|
|
|
|
class FineUploadFileInput(ClearableFileInput):
|
|
template_name = "widgets/fine_uploader.html"
|
|
|
|
def fine_uploader_id(self, name):
|
|
return name + "_" + "fine_uploader"
|
|
|
|
def get_context(self, name, value, attrs):
|
|
context = super().get_context(name, value, attrs)
|
|
context["widget"].update(
|
|
{
|
|
"fine_uploader_id": self.fine_uploader_id(name),
|
|
}
|
|
)
|
|
return context
|