112 lines
3.6 KiB
Python
112 lines
3.6 KiB
Python
from fastapi import FastAPI, HTTPException
|
||
from fastapi.middleware.cors import CORSMiddleware
|
||
import os
|
||
import logging
|
||
from dotenv import load_dotenv
|
||
|
||
from app.routers import qcloud_router, dress_router, tryon_router
|
||
from app.utils.config import get_settings
|
||
from app.utils.response import APIResponse
|
||
from app.database import Base, engine
|
||
from app.middleware import ResponseWrapperMiddleware
|
||
from app.exceptions import setup_exception_handlers
|
||
|
||
# 创建数据库表
|
||
Base.metadata.create_all(bind=engine)
|
||
|
||
# 加载环境变量
|
||
load_dotenv(dotenv_path=".env")
|
||
|
||
# 配置日志
|
||
logging.basicConfig(
|
||
level=logging.INFO,
|
||
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
|
||
)
|
||
logger = logging.getLogger(__name__)
|
||
|
||
app = FastAPI(
|
||
title="AI-Dressing API",
|
||
description="基于 DashScope 的 AI 服务 API",
|
||
version="0.1.0",
|
||
)
|
||
|
||
# 添加 CORS 中间件
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=["*"], # 在生产环境中，应该指定确切的域名
|
||
allow_credentials=True,
|
||
allow_methods=["*"],
|
||
allow_headers=["*"],
|
||
)
|
||
|
||
# 添加响应包装中间件
|
||
app.add_middleware(ResponseWrapperMiddleware)
|
||
|
||
# 设置异常处理
|
||
setup_exception_handlers(app)
|
||
|
||
# 注册路由
|
||
app.include_router(qcloud_router.router, prefix="/api/qcloud", tags=["腾讯云"])
|
||
app.include_router(dress_router.router, prefix="/api/dresses", tags=["服装"])
|
||
app.include_router(tryon_router.router, prefix="/api/tryons", tags=["试穿"])
|
||
|
||
@app.get("/", tags=["健康检查"])
|
||
async def root():
|
||
"""API 根端点"""
|
||
return APIResponse.ok(message="服务运行中")
|
||
|
||
@app.get("/health", tags=["健康检查"])
|
||
async def health_check():
|
||
"""健康检查端点，用于Docker容器健康监控"""
|
||
health_data = {"checks": {}}
|
||
health_status = "healthy"
|
||
health_message = "服务运行正常"
|
||
|
||
# 检查数据库连接
|
||
try:
|
||
from app.database import SessionLocal
|
||
db = SessionLocal()
|
||
db.execute("SELECT 1")
|
||
db.close()
|
||
health_data["checks"]["database"] = {"status": "healthy", "message": "数据库连接正常"}
|
||
except Exception as e:
|
||
health_status = "unhealthy"
|
||
health_message = "服务异常"
|
||
health_data["checks"]["database"] = {"status": "unhealthy", "message": f"数据库连接失败: {str(e)}"}
|
||
|
||
# 检查API密钥
|
||
if os.getenv("DASHSCOPE_API_KEY"):
|
||
health_data["checks"]["dashscope"] = {"status": "configured", "message": "DashScope API密钥已配置"}
|
||
else:
|
||
health_data["checks"]["dashscope"] = {"status": "missing", "message": "DashScope API密钥未配置"}
|
||
|
||
if os.getenv("QCLOUD_SECRET_ID") and os.getenv("QCLOUD_SECRET_KEY"):
|
||
health_data["checks"]["qcloud"] = {"status": "configured", "message": "腾讯云凭证已配置"}
|
||
else:
|
||
health_data["checks"]["qcloud"] = {"status": "missing", "message": "腾讯云凭证未配置"}
|
||
|
||
if health_status == "healthy":
|
||
return APIResponse.ok(data=health_data, message=health_message)
|
||
else:
|
||
return APIResponse.error(message=health_message, code=503, data=health_data)
|
||
|
||
@app.get("/info", tags=["服务信息"])
|
||
async def get_info():
|
||
"""获取服务基本信息"""
|
||
settings = get_settings()
|
||
info_data = {
|
||
"app_name": "AI-Dressing API",
|
||
"version": "0.1.0",
|
||
"debug_mode": settings.debug,
|
||
}
|
||
return APIResponse.ok(data=info_data, message="服务信息获取成功")
|
||
|
||
if __name__ == "__main__":
|
||
import uvicorn
|
||
settings = get_settings()
|
||
uvicorn.run(
|
||
"app.main:app",
|
||
host=settings.host,
|
||
port=settings.port,
|
||
reload=settings.debug,
|
||
) |