dashboard/backend/main.py

from fastapi import FastAPI, Depends, Request
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import JSONResponse
import os
import logging
from dotenv import load_dotenv

from database import engine, Base
from routes import brands_router, models_router, ota_router
from routes.auth import router as auth_router
from security import get_current_user

# Load environment variables
load_dotenv(override=True)

# Configure logging
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
    handlers=[
        logging.StreamHandler(),  # 输出到控制台
        logging.FileHandler('logs/app.log', encoding='utf-8')  # 输出到文件
    ]
)
logger = logging.getLogger(__name__)

# Create database tables (only if tables don't exist)
try:
    logger.info("Creating database tables...")
    Base.metadata.create_all(bind=engine)
    logger.info("Database tables created successfully")
except Exception as e:
    logger.error(f"Warning: Could not create tables: {e}")
    logger.error("Continuing without table creation...")

# Create FastAPI app
app = FastAPI(
    title=os.getenv("APP_NAME", "Audio Dashboard API"),
    description="耳机品牌与型号管理平台后端 API",
    version="1.0.0"
)

# 限制请求体最大 8MB
MAX_BODY_SIZE = 8 * 1024 * 1024  # 8MB


@app.middleware("http")
async def limit_body_size(request: Request, call_next):
    content_length = request.headers.get("content-length")
    if content_length and int(content_length) > MAX_BODY_SIZE:
        return JSONResponse(
            status_code=413,
            content={"detail": "上传文件大小超过 8MB 限制"},
        )
    response = await call_next(request)
    return response


# Configure CORS
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],  # 生产环境应该限制具体域名
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# Include routers（业务接口需登录；登录接口除外）
app.include_router(auth_router)
app.include_router(brands_router, dependencies=[Depends(get_current_user)])
app.include_router(models_router, dependencies=[Depends(get_current_user)])
app.include_router(ota_router, dependencies=[Depends(get_current_user)])


@app.get("/")
def root():
    """根路径"""
    return {
        "message": "欢迎使用 Audio Dashboard API",
        "docs": "/docs",
        "redoc": "/redoc"
    }


@app.get("/health")
def health_check():
    """健康检查1"""
    return {"status": "healthy"}


if __name__ == "__main__":
    import uvicorn
    uvicorn.run(
        "main:app",
        host="0.0.0.0",
        port=8083,
        reload=False
    )