第6章 - 完整代码

"""
学生管理系统 - 主程序入口
"""
import time
from fastapi import FastAPI, Request, HTTPException
from fastapi.middleware.cors import CORSMiddleware
from fastapi.exceptions import RequestValidationError

from app.config import settings
from app.database import engine, Base
from app.models import Class, Student, Score
from app.api import classes, students, scores, statistics
from app.utils.exceptions import (
    http_exception_handler,
    validation_exception_handler,
    general_exception_handler
)
from app.utils.logger import log_info

# 创建数据库表
Base.metadata.create_all(bind=engine)

# 创建 FastAPI 应用
app = FastAPI(
    title=settings.PROJECT_NAME,
    description=settings.PROJECT_DESCRIPTION,
    version=settings.PROJECT_VERSION,
    docs_url="/docs",
    redoc_url="/redoc",
    openapi_tags=[
        {"name": "班级管理", "description": "班级的增删改查操作"},
        {"name": "学生管理", "description": "学生的增删改查操作"},
        {"name": "成绩管理", "description": "成绩的录入和查询"},
        {"name": "数据统计", "description": "各类统计数据"},
    ]
)

# 配置 CORS
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# 注册异常处理器
app.add_exception_handler(HTTPException, http_exception_handler)
app.add_exception_handler(RequestValidationError, validation_exception_handler)
app.add_exception_handler(Exception, general_exception_handler)

# 请求日志中间件
@app.middleware("http")
async def log_requests(request: Request, call_next):
    start_time = time.time()
    response = await call_next(request)
    process_time = time.time() - start_time
    log_info(f"{request.method} {request.url.path} - {response.status_code} - {process_time:.3f}s")
    return response

# 注册路由
app.include_router(classes.router, prefix=settings.API_PREFIX, tags=["班级管理"])
app.include_router(students.router, prefix=settings.API_PREFIX, tags=["学生管理"])
app.include_router(scores.router, prefix=settings.API_PREFIX, tags=["成绩管理"])
app.include_router(statistics.router, prefix=settings.API_PREFIX, tags=["数据统计"])

@app.get("/", tags=["系统"])
def root():
    """系统首页"""
    return {
        "message": "欢迎使用学生管理系统",
        "version": settings.PROJECT_VERSION,
        "docs": "/docs",
        "redoc": "/redoc"
    }

@app.get("/health", tags=["系统"])
def health_check():
    """健康检查"""
    return {"status": "healthy", "version": settings.PROJECT_VERSION}