|
|
@@ -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})
|
|
|
|