diff --git a/main.py b/main.py new file mode 100644 index 0000000..366feb4 --- /dev/null +++ b/main.py @@ -0,0 +1,179 @@ +import datetime +import uuid +import random +import asyncio +from pathlib import Path + +from fastapi import FastAPI, Depends, UploadFile, Form, File, HTTPException +from starlette.responses import HTMLResponse, FileResponse +from starlette.staticfiles import StaticFiles + +from sqlalchemy import or_, select, update, delete +from sqlalchemy.ext.asyncio.session import AsyncSession + +import settings +from database import get_session, Codes, init_models, engine +from storage import STORAGE_ENGINE +from depends import admin_required, IPRateLimit + +app = FastAPI(debug=settings.DEBUG) + +DATA_ROOT = Path(settings.DATA_ROOT) +if not DATA_ROOT.exists(): + DATA_ROOT.mkdir(parents=True) + +STATIC_URL = settings.STATIC_URL +app.mount(STATIC_URL, StaticFiles(directory=DATA_ROOT), name="static") + +storage = STORAGE_ENGINE[settings.STORAGE_ENGINE]() + + +@app.on_event('startup') +async def startup(): + await init_models() + asyncio.create_task(delete_expire_files()) + + +index_html = open('templates/index.html', 'r', encoding='utf-8').read() \ + .replace('{{title}}', settings.TITLE) \ + .replace('{{description}}', settings.DESCRIPTION) \ + .replace('{{keywords}}', settings.KEYWORDS) +admin_html = open('templates/admin.html', 'r', encoding='utf-8').read() \ + .replace('{{title}}', settings.TITLE) \ + .replace('{{description}}', settings.DESCRIPTION) \ + .replace('{{keywords}}', settings.KEYWORDS) + +ip_limit = IPRateLimit() + + +async def delete_expire_files(): + while True: + async with AsyncSession(engine, expire_on_commit=False) as s: + query = select(Codes).where(or_(Codes.exp_time < datetime.datetime.now(), Codes.count == 0)) + exps = (await s.execute(query)).scalars().all() + files = [{'type': old.type, 'text': old.text} for old in exps] + await storage.delete_files(files) + exps_ids = [exp.id for exp in exps] + query = delete(Codes).where(Codes.id.in_(exps_ids)) + await s.execute(query) + await s.commit() + await asyncio.sleep(random.randint(60, 300)) + + +async def get_code(s: AsyncSession): + code = random.randint(10000, 99999) + while (await s.execute(select(Codes.id).where(Codes.code == code))).scalar(): + code = random.randint(10000, 99999) + return str(code) + + +@app.get(f'/{settings.ADMIN_ADDRESS}') +async def admin(): + return HTMLResponse(admin_html) + + +@app.post(f'/{settings.ADMIN_ADDRESS}', dependencies=[Depends(admin_required)]) +async def admin_post(s: AsyncSession = Depends(get_session)): + query = select(Codes) + codes = (await s.execute(query)).scalars().all() + return {'detail': '查询成功', 'data': codes} + + +@app.delete(f'/{settings.ADMIN_ADDRESS}', dependencies=[Depends(admin_required)]) +async def admin_delete(code: str, s: AsyncSession = Depends(get_session)): + query = select(Codes).where(Codes.code == code) + file = (await s.execute(query)).scalars().first() + await storage.delete_file({'type': file.type, 'text': file.text}) + await s.delete(file) + await s.commit() + return {'detail': '删除成功'} + + +@app.get('/') +async def index(): + return HTMLResponse(index_html) + + +@app.get('/select') +async def get_file(code: str, s: AsyncSession = Depends(get_session)): + query = select(Codes).where(Codes.code == code) + info = (await s.execute(query)).scalars().first() + if not info: + raise HTTPException(status_code=404, detail="口令不存在") + if info.type == 'text': + return {'detail': '查询成功', 'data': info.text} + else: + filepath = await storage.get_filepath(info.text) + return FileResponse(filepath, filename=info.name) + + +@app.post('/') +async def index(code: str, ip: str = Depends(ip_limit), s: AsyncSession = Depends(get_session)): + query = select(Codes).where(Codes.code == code) + info = (await s.execute(query)).scalars().first() + if not info: + error_count = settings.ERROR_COUNT - ip_limit.add_ip(ip) + raise HTTPException(status_code=404, detail=f"取件码错误，错误{error_count}次将被禁止10分钟") + if info.exp_time < datetime.datetime.now() or info.count == 0: + await storage.delete_file({'type': info.type, 'text': info.text}) + await s.delete(info) + await s.commit() + raise HTTPException(status_code=404, detail="取件码已过期，请联系寄件人") + await s.execute(update(Codes).where(Codes.id == info.id).values(count=info.count - 1)) + await s.commit() + if info.type != 'text': + info.text = f'/select?code={code}' + return { + 'detail': '取件成功，请点击"取"查看', + 'data': {'type': info.type, 'text': info.text, 'name': info.name, 'code': info.code} + } + + +@app.post('/share') +async def share(text: str = Form(default=None), style: str = Form(default='2'), value: int = Form(default=1), + file: UploadFile = File(default=None), s: AsyncSession = Depends(get_session)): + code = await get_code(s) + if style == '2': + if value > 7: + raise HTTPException(status_code=400, detail="最大有效天数为7天") + exp_time = datetime.datetime.now() + datetime.timedelta(days=value) + exp_count = -1 + elif style == '1': + if value < 1: + raise HTTPException(status_code=400, detail="最小有效次数为1次") + exp_time = datetime.datetime.now() + datetime.timedelta(days=1) + exp_count = value + else: + exp_time = datetime.datetime.now() + datetime.timedelta(days=1) + exp_count = -1 + key = uuid.uuid4().hex + if file: + file_bytes = await file.read() + size = len(file_bytes) + if size > settings.FILE_SIZE_LIMIT: + raise HTTPException(status_code=400, detail="文件过大") + _text, _type, name = await storage.save_file(file, file_bytes, key), file.content_type, file.filename + else: + size, _text, _type, name = len(text), text, 'text', '文本分享' + info = Codes( + code=code, + text=_text, + size=size, + type=_type, + name=name, + count=exp_count, + exp_time=exp_time, + key=key + ) + s.add(info) + await s.commit() + return { + 'detail': '分享成功，请点击文件箱查看取件码', + 'data': {'code': code, 'key': key, 'name': name, 'text': _text} + } + + +if __name__ == '__main__': + import uvicorn + + uvicorn.run('main:app', host='0.0.0.0', port=settings.PORT, debug=settings.DEBUG) diff --git a/templates/admin.html b/templates/admin.html new file mode 100644 index 0000000..6d2be2b --- /dev/null +++ b/templates/admin.html @@ -0,0 +1,111 @@ + + + + + + + + + 后台管理-{{title}} + + + + + + + + +
+ + +   + + + + + + + +
+
取件码：${ file.code }
+
文件名：${ file.name }
+
次   数：${ file.count }
+
到   期：${ file.exp_time.slice(0,19) }
+
+ 内   容：${ file.text } +
+
+ 链   接： + 点击下载 +
+
+ + + 删除 + + + + + +   + +
+ + 登录 + +
+
+ + + + + \ No newline at end of file diff --git a/templates/index.html b/templates/index.html new file mode 100644 index 0000000..6fc228e --- /dev/null +++ b/templates/index.html @@ -0,0 +1,318 @@ + + + + + + + + + {{title}} + + + + + + + + + + +
+ + + + + + + + 1 + + + 2 + + + 3 + + + 4 + + + 5 + + + 6 + + + 7 + + + 8 + + + 9 + + + 寄 + + + 0 + + + 取 + + + + + + + + + + + + 次 + 天 + + + 文件 + 文本 + + + +
将文字、文件拖、粘贴到此处，或点击上传
+
天数<7或限制次数（24h后删除）
+ + + +
+ 去取件 + 取件码 + 寄出去 +
+ + + + + +
+
取件码：${ file.code }
+
文件名：${ file.name }
+
+ 内   容：${ file.text } +
+
+ 链   接： + 点击下载 +
+
+ + + + + + + + + + + + ${ file.name } + + + 取件码：

${ file.code }

+ + + + + 二维码 + + + + + +
+ FileCodeBox +
+
+ + + + \ No newline at end of file