From 70b931394fa8d13f35b35436834a0cd07f6893e6 Mon Sep 17 00:00:00 2001 From: lan-air Date: Sun, 11 Dec 2022 14:40:14 +0800 Subject: [PATCH] =?UTF-8?q?=E9=9A=90=E8=97=8F=E7=9C=9F=E5=AE=9E=E6=96=87?= =?UTF-8?q?=E4=BB=B6=E5=9C=B0=E5=9D=80?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index 14e6c2f..64723a1 100644 --- a/main.py +++ b/main.py @@ -6,7 +6,7 @@ from fastapi import FastAPI, Depends, UploadFile, Form, File from sqlalchemy import or_ from sqlalchemy.orm import Session from starlette.requests import Request -from starlette.responses import HTMLResponse +from starlette.responses import HTMLResponse, FileResponse import random from starlette.staticfiles import StaticFiles @@ -137,6 +137,18 @@ def ip_error(ip): return ip_info['count'] +@app.get('/file') +async def get_file(code: str, db: Session = Depends(get_db)): + file = db.query(database.Codes).filter(database.Codes.code == code).first() + if file: + if file.type == 'text': + return {'code': code, 'msg': '查询成功', 'data': file.text} + else: + return FileResponse('.' + file.text) + else: + return {'code': 404, 'msg': '口令不存在'} + + @app.post('/') async def index(request: Request, code: str, db: Session = Depends(get_db)): ip = request.client.host @@ -152,6 +164,8 @@ async def index(request: Request, code: str, db: Session = Depends(get_db)): return {'code': 404, 'msg': '取件码已过期，请联系寄件人'} info.count -= 1 db.commit() + if info.type != 'text': + info.text = f'/file?code={code}' return { 'code': 200, 'msg': '取件成功，请点击"取"查看',