You've already forked DataMate
2f7341dc1f
refactor: Reorganize datamate-python (previously label-studio-adapter) into a DDD style structure.
111 lines
3.4 KiB
Python
111 lines
3.4 KiB
Python
from fastapi import FastAPI, Request, HTTPException, status
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.exceptions import RequestValidationError
|
|
from starlette.exceptions import HTTPException as StarletteHTTPException
|
|
|
|
from contextlib import asynccontextmanager
|
|
from typing import Dict, Any
|
|
from sqlalchemy import text
|
|
|
|
from .core.config import settings
|
|
from .core.logging import setup_logging, get_logger
|
|
from .db.session import engine, AsyncSessionLocal
|
|
from .module.shared.schema import StandardResponse
|
|
from .module import router
|
|
from .exception import (
|
|
starlette_http_exception_handler,
|
|
fastapi_http_exception_handler,
|
|
validation_exception_handler,
|
|
general_exception_handler
|
|
)
|
|
|
|
# 设置日志
|
|
setup_logging()
|
|
logger = get_logger(__name__)
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
"""应用程序生命周期管理"""
|
|
|
|
# 启动时初始化
|
|
logger.info("DataMate Python Backend starting...")
|
|
# 数据库连接验证
|
|
try:
|
|
async with AsyncSessionLocal() as session:
|
|
await session.execute(text("SELECT 1"))
|
|
logger.info("Database connection validated successfully.")
|
|
except Exception as e:
|
|
logger.error(f"Database connection validation failed: {e}")
|
|
logger.debug(f"Connection details: {settings.computed_database_url}")
|
|
raise
|
|
|
|
yield
|
|
|
|
# 关闭时清理
|
|
logger.info("DataMate Python Backend shutting down ...")
|
|
|
|
# 创建FastAPI应用
|
|
app = FastAPI(
|
|
title=settings.app_name,
|
|
description=settings.app_description,
|
|
version=settings.app_version,
|
|
debug=settings.debug,
|
|
lifespan=lifespan
|
|
)
|
|
|
|
# 配置CORS中间件
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=settings.allowed_origins,
|
|
allow_credentials=True,
|
|
allow_methods=settings.allowed_methods,
|
|
allow_headers=settings.allowed_headers,
|
|
)
|
|
|
|
# 注册路由
|
|
app.include_router(router)
|
|
|
|
logger.debug("Registered routes: %s", [getattr(r, "path", None) for r in app.routes])
|
|
|
|
# 注册全局异常处理器
|
|
app.add_exception_handler(StarletteHTTPException, starlette_http_exception_handler) # type: ignore
|
|
app.add_exception_handler(HTTPException, fastapi_http_exception_handler) # type: ignore
|
|
app.add_exception_handler(RequestValidationError, validation_exception_handler) # type: ignore
|
|
app.add_exception_handler(Exception, general_exception_handler)
|
|
|
|
# 测试端点：验证异常处理
|
|
@app.get("/test-404", include_in_schema=False)
|
|
async def test_404():
|
|
"""测试404异常处理"""
|
|
raise HTTPException(status_code=404, detail="Test 404 error")
|
|
|
|
@app.get("/test-500", include_in_schema=False)
|
|
async def test_500():
|
|
"""测试500异常处理"""
|
|
raise Exception("Test uncaught exception")
|
|
|
|
# 根路径重定向到文档
|
|
@app.get("/", response_model=StandardResponse[Dict[str, Any]], include_in_schema=False)
|
|
async def root():
|
|
"""根路径，返回服务信息"""
|
|
return StandardResponse(
|
|
code=200,
|
|
message="success",
|
|
data={
|
|
"message": f"{settings.app_name} is running",
|
|
"version": settings.app_version,
|
|
"docs_url": "/docs",
|
|
"label_studio_url": settings.label_studio_base_url
|
|
}
|
|
)
|
|
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
|
|
uvicorn.run(
|
|
"app.main:app",
|
|
host=settings.host,
|
|
port=settings.port,
|
|
reload=settings.debug,
|
|
log_level=settings.log_level.lower()
|
|
) |