style: code format
This commit is contained in:
@@ -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
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user