于小丘
209cb24ab4
- Implement PhysicalFile model to manage physical file references and reference counting. - Create Policy model with associated options and group links for storage policies. - Introduce Redeem and Report models for handling redeem codes and reports. - Add Settings model for site configuration and user settings management. - Develop Share model for sharing objects with unique codes and associated metadata. - Implement SourceLink model for managing download links associated with objects. - Create StoragePack model for managing user storage packages. - Add Tag model for user-defined tags with manual and automatic types. - Implement Task model for managing background tasks with status tracking. - Develop User model with comprehensive user management features including authentication. - Introduce UserAuthn model for managing WebAuthn credentials. - Create WebDAV model for managing WebDAV accounts associated with users.
71 lines
2.0 KiB
Python
71 lines
2.0 KiB
Python
from typing import NoReturn
|
|
|
|
from fastapi import FastAPI, Request
|
|
|
|
from utils.conf import appmeta
|
|
from utils.http.http_exceptions import raise_internal_error
|
|
from utils.lifespan import lifespan
|
|
from sqlmodels.database_connection import DatabaseManager
|
|
from sqlmodels.migration import migration
|
|
from utils import JWT
|
|
from routers import router
|
|
from service.redis import RedisManager
|
|
from loguru import logger as l
|
|
|
|
async def _init_db() -> None:
|
|
"""初始化数据库连接引擎"""
|
|
await DatabaseManager.init(appmeta.database_url, debug=appmeta.debug)
|
|
|
|
# 添加初始化数据库启动项
|
|
lifespan.add_startup(_init_db)
|
|
lifespan.add_startup(migration)
|
|
lifespan.add_startup(JWT.load_secret_key)
|
|
lifespan.add_startup(RedisManager.connect)
|
|
|
|
# 添加关闭项
|
|
lifespan.add_shutdown(DatabaseManager.close)
|
|
lifespan.add_shutdown(RedisManager.disconnect)
|
|
|
|
# 创建应用实例并设置元数据
|
|
app = FastAPI(
|
|
title=appmeta.APP_NAME,
|
|
summary=appmeta.summary,
|
|
description=appmeta.description,
|
|
version=appmeta.BackendVersion,
|
|
license_info=appmeta.license_info,
|
|
lifespan=lifespan.lifespan,
|
|
debug=appmeta.debug,
|
|
openapi_url="/openapi.json" if appmeta.debug else None,
|
|
)
|
|
# 添加跨域 CORS 中间件,仅在调试模式下启用,以允许所有来源访问 API
|
|
if appmeta.debug:
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
@app.exception_handler(Exception)
|
|
async def handle_unexpected_exceptions(
|
|
request: Request,
|
|
exc: Exception
|
|
) -> NoReturn:
|
|
"""
|
|
捕获所有未经处理的 FastAPI 异常,防止敏感信息泄露。
|
|
"""
|
|
l.exception(exc)
|
|
l.error(f"An unhandled exception occurred for request: {request.method} {request.url.path}")
|
|
|
|
raise_internal_error()
|
|
|
|
# 挂载路由
|
|
app.include_router(router)
|
|
|
|
# 防止直接运行 main.py
|
|
if __name__ == "__main__":
|
|
l.error("请用 fastapi ['dev', 'run'] 命令启动服务")
|
|
exit(1)
|
|
|