完成最简单的接收文件功能

This commit is contained in:
dmy 2020-02-18 20:37:49 +08:00
parent f6aba7c7ed
commit 3bb2c6862d
1 changed files with 33 additions and 0 deletions

33
src/server/app.py Normal file
View File

@ -0,0 +1,33 @@
from flask import Flask
from flask import request
from flask_cors import CORS
import util
import db.model
from db.model import Pic
from db.connector import Session
app = Flask(__name__)
CORS(app)
@app.route("/upload", methods=["POST", "GET"])
def upload():
# app.logger.debug("Got upload post.")
if request.method != "POST":
return "{status:'Err',msg:'only post is supported.'}"
pic_storage_dir='/home/dmy/doc/pic_storage'
f=request.files['up']
file_binary=f.read()
md5=util.md5(file_binary)
#先检查是否已经有图片了
session=Session()
# test_existed_pic=Pic(md5=md5)
if session.query(Pic.md5).filter(Pic.md5==md5).first() is not None:
app.logger.debug('pic with md5 {md5} esisted.'.format(md5=md5))
return '{ status:"succeed" }'
save_path='{pic_storage_dir}/{md5}'.format(pic_storage_dir=pic_storage_dir,md5=md5)
utc_now=util.utc_now()
db.model.add_pic(md5=md5,path=save_path,time=utc_now)
f.save(save_path)
return '{ status:"succeed" }'