122 lines
4.3 KiB
Python
122 lines
4.3 KiB
Python
|
# main0814.py
|
|||
|
|
# 项目入口文件：创建 FastAPI 应用、注册路由、创建数据库表
|
|||
|
|
import os
|
|||
|
|
from fastapi import FastAPI
|
|||
|
|
from pathlib import Path
|
|||
|
|
from fastapi.responses import FileResponse, HTMLResponse, JSONResponse
|
|||
|
|
from fastapi.staticfiles import StaticFiles
|
|||
|
|
from sqlalchemy.exc import IntegrityError, SQLAlchemyError
|
|||
|
|
from api import workspace_api, ai_api
|
|||
|
|
from starlette.middleware.cors import CORSMiddleware
|
|||
|
|
|
|||
|
|
from api import student_api, advisor_api, employment_api, score_api, class_api, teacher_api
|
|||
|
|
from init_db import init_all, drop_tables
|
|||
|
|
from database import engine
|
|||
|
|
from sqlalchemy import text
|
|||
|
|
from service.web_auth import install_web_auth
|
|||
|
|
|
|||
|
|
# 2. 创建 FastAPI 实例
|
|||
|
|
app = FastAPI(
|
|||
|
|
title="FastAPI + SQLAlchemy 分层架构（MySQL）",
|
|||
|
|
description="用户管理示例，演示分层架构和 MySQL 集成",
|
|||
|
|
version="1.0.0"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 3. 添加跨域中间件（允许前端跨域请求）
|
|||
|
|
app.add_middleware(
|
|||
|
|
CORSMiddleware,
|
|||
|
|
allow_origins=[value.strip() for value in os.getenv("CORS_ORIGINS", "*").split(",") if value.strip()],
|
|||
|
|
allow_credentials=True,
|
|||
|
|
allow_methods=["*"],
|
|||
|
|
allow_headers=["*"],
|
|||
|
|
)
|
|||
|
|
install_web_auth(app)
|
|||
|
|
|
|||
|
|
# 4. 注册子路由
|
|||
|
|
# prefix 为路由前缀，所有用户接口都以 /api/users 开头
|
|||
|
|
app.include_router(
|
|||
|
|
student_api.router,
|
|||
|
|
prefix="/api/studentsManagement",
|
|||
|
|
tags=["学生管理 (文涛,浩霖)"]
|
|||
|
|
)
|
|||
|
|
#
|
|||
|
|
app.include_router(
|
|||
|
|
class_api.router,
|
|||
|
|
prefix="/api/studentsManagement",
|
|||
|
|
tags=["班级管理 (薛炎)"]
|
|||
|
|
)
|
|||
|
|
#
|
|||
|
|
app.include_router(
|
|||
|
|
teacher_api.router,
|
|||
|
|
prefix="/api/studentsManagement",
|
|||
|
|
tags=["老师管理 (广龙)"]
|
|||
|
|
)
|
|||
|
|
#
|
|||
|
|
app.include_router(
|
|||
|
|
advisor_api.router,
|
|||
|
|
prefix="/api/studentsManagement",
|
|||
|
|
tags=["顾问管理 (子镜)"]
|
|||
|
|
)
|
|||
|
|
#
|
|||
|
|
app.include_router(
|
|||
|
|
score_api.router,
|
|||
|
|
prefix="/api/studentsManagement",
|
|||
|
|
tags=["成绩管理 (广湘)"]
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
app.include_router(
|
|||
|
|
employment_api.router,
|
|||
|
|
prefix="/api/studentsManagement",
|
|||
|
|
tags=["就业管理 (周兴)"]
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 5. 前端及聚合查询接口
|
|||
|
|
app.include_router(workspace_api.router, prefix="/api/studentsManagement")
|
|||
|
|
app.include_router(ai_api.router, prefix="/api/studentsManagement")
|
|||
|
|
FRONTEND_DIR = Path(__file__).resolve().parent / "frontend"
|
|||
|
|
app.mount("/static", StaticFiles(directory=FRONTEND_DIR), name="frontend")
|
|||
|
|
|
|||
|
|
|
|||
|
|
@app.exception_handler(IntegrityError)
|
|||
|
|
async def integrity_error(request, error):
|
|||
|
|
return JSONResponse(status_code=409, content={"detail": "保存失败：学号、手机号、名称或学生就业记录已存在，或关联记录不存在。请检查后重试。"})
|
|||
|
|
|
|||
|
|
|
|||
|
|
@app.exception_handler(SQLAlchemyError)
|
|||
|
|
async def database_error(request, error):
|
|||
|
|
return JSONResponse(status_code=503, content={"detail": "数据库暂时不可用，请检查 MySQL 服务和表结构。"})
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 根路径直接打开前端页面，API 文档仍位于 /docs。
|
|||
|
|
@app.get("/")
|
|||
|
|
async def root():
|
|||
|
|
if os.getenv("AUTH_ENABLED", "").lower() in {"1", "true", "yes", "on"}:
|
|||
|
|
html = (FRONTEND_DIR / "index.html").read_text(encoding="utf-8")
|
|||
|
|
actions = '''<button type="button" class="button subtle" id="sign-out">退出登录</button>
|
|||
|
|
<script>document.getElementById('sign-out').addEventListener('click',async function(){
|
|||
|
|
this.disabled=true;try{const base=new URL('.',document.baseURI);
|
|||
|
|
const response=await fetch(new URL('auth/logout',base),{method:'POST',credentials:'same-origin'});
|
|||
|
|
if(!response.ok)throw new Error();location.replace(new URL('login',base).href);
|
|||
|
|
}catch(error){this.disabled=false;this.textContent='退出失败，请重试';}});</script>'''
|
|||
|
|
return HTMLResponse(html.replace("<!--AUTH_ACTIONS-->", actions), headers={"Cache-Control": "no-store"})
|
|||
|
|
return FileResponse(FRONTEND_DIR / "index.html")
|
|||
|
|
|
|||
|
|
|
|||
|
|
@app.get("/healthz", include_in_schema=False)
|
|||
|
|
def healthz():
|
|||
|
|
with engine.connect() as connection:
|
|||
|
|
connection.execute(text("SELECT 1"))
|
|||
|
|
return {"status": "ok"}
|
|||
|
|
|
|||
|
|
# 6. 如果直接运行此文件，启动 uvicorn 服务器
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
# drop_tables()
|
|||
|
|
init_all(True, False)
|
|||
|
|
import uvicorn
|
|||
|
|
uvicorn.run(
|
|||
|
|
"main:app", # 指定应用位置（模块名:应用变量名）
|
|||
|
|
host=os.getenv("APP_HOST", "127.0.0.1"), # 默认仅本机访问，不依赖 Wi-Fi / 局域网 IP
|
|||
|
|
port=int(os.getenv("APP_PORT", "8004")), # 可通过环境变量调整端口
|
|||
|
|
reload=True # 开发模式，代码变动自动重启
|
|||
|
|
)
|