增加文件上传功能

utils/views.py

@@ -1,6 +1,6 @@
 import os
 from django.conf import settings
-from account.serializers import ImageUploadForm
+from account.serializers import ImageUploadForm, FileUploadForm
 from utils.shortcuts import rand_str
 from utils.api import CSRFExemptAPIView
 import logging
@@ -35,10 +35,41 @@ class SimditorImageUploadAPIView(CSRFExemptAPIView):
         except IOError as e:
             logger.error(e)
             return self.response({
-                "success": True,
+                "success": False,
                 "msg": "Upload Error",
-                "file_path": f"{settings.UPLOAD_PREFIX}/{img_name}"})
+                "file_path": ""})
         return self.response({
             "success": True,
             "msg": "Success",
             "file_path": f"{settings.UPLOAD_PREFIX}/{img_name}"})
+
+
+class SimditorFileUploadAPIView(CSRFExemptAPIView):
+    request_parsers = ()
+
+    def post(self, request):
+        form = FileUploadForm(request.POST, request.FILES)
+        if form.is_valid():
+            file = form.cleaned_data["file"]
+        else:
+            return self.response({
+                "success": False,
+                "msg": "Upload failed"
+            })
+
+        suffix = os.path.splitext(file.name)[-1].lower()
+        file_name = rand_str(10) + suffix
+        try:
+            with open(os.path.join(settings.UPLOAD_DIR, file_name), "wb") as f:
+                for chunk in file:
+                    f.write(chunk)
+        except IOError as e:
+            logger.error(e)
+            return self.response({
+                "success": False,
+                "msg": "Upload Error"})
+        return self.response({
+            "success": True,
+            "msg": "Success",
+            "file_path": f"{settings.UPLOAD_PREFIX}/{file_name}",
+            "file_name": file.name})