diff --git a/main.py b/main.py index 5d73d40..bc53ca4 100644 --- a/main.py +++ b/main.py @@ -12,7 +12,7 @@ from tortoise.contrib.fastapi import register_tortoise from apps.base.views import share_api from apps.admin.views import admin_api -from core.settings import data_root, settings +from core.settings import data_root, settings, BASE_DIR from core.tasks import delete_expire_files from core.utils import max_save_times_desc @@ -31,13 +31,13 @@ app.add_middleware( async def assets(file_path: str): if settings.max_save_seconds > 0: if re.match(r'SendView-[\d|a-f|A-F]+\.js', file_path): - with open(f'./fcb-fronted/dist/assets/{file_path}', 'r', encoding='utf-8') as f: + with open(BASE_DIR / f'./fcb-fronted/dist/assets/{file_path}', 'r', encoding='utf-8') as f: # 删除永久保存选项 content = f.read() content = content.replace('_(c,{label:e(r)("send.expireData.forever"),value:"forever"},null,8,["label"]),', '') return HTMLResponse(content=content, media_type='text/javascript') if re.match(r'index-[\d|a-f|A-F]+\.js', file_path): - with open(f'./fcb-fronted/dist/assets/{file_path}', 'r', encoding='utf-8') as f: + with open(BASE_DIR / f'./fcb-fronted/dist/assets/{file_path}', 'r', encoding='utf-8') as f: # 更改本文描述 desc_zh, desc_en = await max_save_times_desc(settings.max_save_seconds) content = f.read()