style: code format

This commit is contained in:
Lan
2025-02-15 22:00:53 +08:00
parent 802d2551ff
commit 1ff1d6c695
14 changed files with 404 additions and 311 deletions
+56 -39
View File
@@ -26,10 +26,10 @@ from tortoise import Tortoise
async def init_db():
await Tortoise.init(
db_url=f'sqlite://{data_root}/filecodebox.db',
modules={'models': ['apps.base.models']},
db_url=f"sqlite://{data_root}/filecodebox.db",
modules={"models": ["apps.base.models"]},
use_tz=False,
timezone="Asia/Shanghai"
timezone="Asia/Shanghai",
)
await Tortoise.generate_schemas()
@@ -41,7 +41,11 @@ async def lifespan(app: FastAPI):
# 加载配置
await load_config()
app.mount('/assets', StaticFiles(directory=f'./{settings.themesSelect}/assets'), name="assets")
app.mount(
"/assets",
StaticFiles(directory=f"./{settings.themesSelect}/assets"),
name="assets",
)
# 启动后台任务
task = asyncio.create_task(delete_expire_files())
@@ -56,14 +60,18 @@ async def lifespan(app: FastAPI):
async def load_config():
user_config, _ = await KeyValue.get_or_create(key='settings', defaults={'value': DEFAULT_CONFIG})
await KeyValue.update_or_create(key='sys_start', defaults={'value': int(time.time() * 1000)})
user_config, _ = await KeyValue.get_or_create(
key="settings", defaults={"value": DEFAULT_CONFIG}
)
await KeyValue.update_or_create(
key="sys_start", defaults={"value": int(time.time() * 1000)}
)
settings.user_config = user_config.value
# 更新 ip_limit 配置
ip_limit['error'].minutes = settings.errorMinute
ip_limit['error'].count = settings.errorCount
ip_limit['upload'].minutes = settings.uploadMinute
ip_limit['upload'].count = settings.uploadCount
ip_limit["error"].minutes = settings.errorMinute
ip_limit["error"].count = settings.errorCount
ip_limit["upload"].minutes = settings.uploadMinute
ip_limit["upload"].count = settings.uploadCount
app = FastAPI(lifespan=lifespan)
@@ -80,11 +88,11 @@ app.add_middleware(
register_tortoise(
app,
config={
'connections': {'default': f'sqlite://{data_root}/filecodebox.db'},
'apps': {
'models': {
'models': ['apps.base.models'],
'default_connection': 'default',
"connections": {"default": f"sqlite://{data_root}/filecodebox.db"},
"apps": {
"models": {
"models": ["apps.base.models"],
"default_connection": "default",
},
},
},
@@ -96,39 +104,48 @@ app.include_router(share_api)
app.include_router(admin_api)
@app.get('/')
@app.get("/")
async def index():
return HTMLResponse(
content=open(BASE_DIR / f'{settings.themesSelect}/index.html', 'r', encoding='utf-8').read()
.replace('{{title}}', str(settings.name))
.replace('{{description}}', str(settings.description))
.replace('{{keywords}}', str(settings.keywords))
.replace('{{opacity}}', str(settings.opacity))
.replace('{{background}}', str(settings.background))
, media_type='text/html', headers={'Cache-Control': 'no-cache'})
content=open(
BASE_DIR / f"{settings.themesSelect}/index.html", "r", encoding="utf-8"
)
.read()
.replace("{{title}}", str(settings.name))
.replace("{{description}}", str(settings.description))
.replace("{{keywords}}", str(settings.keywords))
.replace("{{opacity}}", str(settings.opacity))
.replace("{{background}}", str(settings.background)),
media_type="text/html",
headers={"Cache-Control": "no-cache"},
)
@app.get('/robots.txt')
@app.get("/robots.txt")
async def robots():
return HTMLResponse(content=settings.robotsText, media_type='text/plain')
return HTMLResponse(content=settings.robotsText, media_type="text/plain")
@app.post('/')
@app.post("/")
async def get_config():
return APIResponse(detail={
'name': settings.name,
'description': settings.description,
'explain': settings.page_explain,
'uploadSize': settings.uploadSize,
'expireStyle': settings.expireStyle,
'openUpload': settings.openUpload,
'notify_title': settings.notify_title,
'notify_content': settings.notify_content,
'show_admin_address': settings.showAdminAddr,
})
return APIResponse(
detail={
"name": settings.name,
"description": settings.description,
"explain": settings.page_explain,
"uploadSize": settings.uploadSize,
"expireStyle": settings.expireStyle,
"openUpload": settings.openUpload,
"notify_title": settings.notify_title,
"notify_content": settings.notify_content,
"show_admin_address": settings.showAdminAddr,
}
)
if __name__ == '__main__':
if __name__ == "__main__":
import uvicorn
uvicorn.run(app='main:app', host="0.0.0.0", port=settings.port, reload=False, workers=1)
uvicorn.run(
app="main:app", host="0.0.0.0", port=settings.port, reload=False, workers=1
)