Quellcode durchsuchen

upload videos api

lyc8503 vor 4 Jahren
Ursprung
Commit
e225ca7db8

+ 1 - 0
backend_refactor/route/code.py

@@ -0,0 +1 @@
+UPLOAD_INVALID_FORMAT = -10001

+ 6 - 3
backend_refactor/route/upload.py

@@ -4,6 +4,7 @@ import uuid
 from flask import request, Blueprint
 
 from config import UPLOAD_DIR
+from route import code
 from route.util import make_response
 
 from PIL import Image
@@ -33,11 +34,13 @@ def upload(type):
             img.save(open(UPLOAD_DIR + random_name + ".jpg", "wb"))
         except Exception as e:
             logging.exception(e)
-            return make_response(-1, "cannot identify image file.", {})
+            return make_response(code.UPLOAD_INVALID_FORMAT, "cannot identify image file.", {})
     elif type == "video":
-        pass
+        if not file.filename.endswith(".mp4"):
+            return make_response(code.UPLOAD_INVALID_FORMAT, "only mp4 videos are accepted.", {})
+        file.save(UPLOAD_DIR + random_name + ".mp4")
     else:
-        assert False, "未知上传格式"
+        assert False, "unknown upload format"
 
     return make_response(0, "success", {"uuid": random_name})
 

+ 0 - 0
backend_refactor/service/__init__.py


+ 0 - 0
backend_refactor/service/photo_analyze.py


+ 0 - 0
backend_refactor/service/video_analyze.py