Skip to content

Commit f2b5490

Browse files
committed
add:解决select可跳过次数限制,新增sha256加密token,限制取件码+ip+时间999s内取件
1 parent 1f303d3 commit f2b5490

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

main.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from sqlalchemy import select, func, update
77
from sqlalchemy.ext.asyncio import AsyncSession
88
from starlette.requests import Request
9-
from starlette.responses import HTMLResponse, FileResponse
9+
from starlette.responses import HTMLResponse, FileResponse, RedirectResponse
1010
from starlette.staticfiles import StaticFiles
1111
from core.utils import error_ip_limit, upload_ip_limit, get_code, storage, delete_expire_files, get_token
1212
from core.depends import admin_required
@@ -166,6 +166,9 @@ async def get_file(code: str, token: str, ip: str = Depends(error_ip_limit), s:
166166
if info.type == 'text':
167167
return {'detail': '查询成功', 'data': info.text}
168168
# 如果是文件,返回文件
169+
elif storage.NAME != 'filesystem':
170+
# 重定向到文件存储服务器
171+
return RedirectResponse(await storage.get_url(info))
169172
else:
170173
filepath = await storage.get_filepath(info.text)
171174
return FileResponse(filepath, filename=info.name)

0 commit comments

Comments
 (0)