|
|
@@ -4,7 +4,6 @@ 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
|
|
|
@@ -15,11 +14,11 @@ bp = Blueprint("upload", __name__, url_prefix='/')
|
|
|
def upload(type):
|
|
|
# 检查是否正确上传了文件
|
|
|
if 'file' not in request.files:
|
|
|
- return make_response(400, "no file part.", {}, 400)
|
|
|
+ return make_response(400, "no file part.", {})
|
|
|
|
|
|
file = request.files['file']
|
|
|
if file.filename == '':
|
|
|
- return make_response(400, "no selected file.", {}, 400)
|
|
|
+ return make_response(400, "no selected file.", {})
|
|
|
|
|
|
# 随机 uuid 名称
|
|
|
random_name = str(uuid.uuid4())
|
|
|
@@ -34,15 +33,15 @@ def upload(type):
|
|
|
img.save(open(UPLOAD_DIR + random_name + ".jpg", "wb"))
|
|
|
except Exception as e:
|
|
|
logging.exception(e)
|
|
|
- return make_response(code.UPLOAD_INVALID_FORMAT, "cannot identify image file.", {})
|
|
|
+ return make_response(400, "cannot identify image file.", {})
|
|
|
elif type == "video":
|
|
|
if not file.filename.endswith(".mp4"):
|
|
|
- return make_response(code.UPLOAD_INVALID_FORMAT, "only mp4 videos are accepted.", {})
|
|
|
+ return make_response(400, "only mp4 videos are accepted.", {})
|
|
|
file.save(UPLOAD_DIR + random_name + ".mp4")
|
|
|
else:
|
|
|
assert False, "unknown upload format"
|
|
|
|
|
|
- return make_response(0, "success", {"uuid": random_name})
|
|
|
+ return make_response(201, "success", {"uuid": random_name})
|
|
|
|
|
|
|
|
|
@bp.route("/photos", methods=["POST"])
|