74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
"""AirLabs Project —— 主入口"""
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.staticfiles import StaticFiles
|
|
from fastapi.responses import FileResponse
|
|
from database import engine, Base
|
|
from models import User, UserRole, PhaseGroup
|
|
from auth import hash_password
|
|
import os
|
|
|
|
# 创建所有表
|
|
Base.metadata.create_all(bind=engine)
|
|
|
|
app = FastAPI(title="AirLabs Project", version="1.0.0")
|
|
|
|
# CORS
|
|
from config import CORS_ORIGINS
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=CORS_ORIGINS,
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# 注册路由
|
|
from routers.auth import router as auth_router
|
|
from routers.users import router as users_router
|
|
from routers.projects import router as projects_router
|
|
from routers.submissions import router as submissions_router
|
|
from routers.costs import router as costs_router
|
|
from routers.dashboard import router as dashboard_router
|
|
|
|
app.include_router(auth_router)
|
|
app.include_router(users_router)
|
|
app.include_router(projects_router)
|
|
app.include_router(submissions_router)
|
|
app.include_router(costs_router)
|
|
app.include_router(dashboard_router)
|
|
|
|
# 前端静态文件
|
|
frontend_dir = os.path.join(os.path.dirname(os.path.dirname(__file__)), "frontend", "dist")
|
|
if os.path.exists(frontend_dir):
|
|
app.mount("/assets", StaticFiles(directory=os.path.join(frontend_dir, "assets")), name="assets")
|
|
|
|
@app.get("/{full_path:path}")
|
|
async def serve_frontend(full_path: str):
|
|
file_path = os.path.join(frontend_dir, full_path)
|
|
if os.path.isfile(file_path):
|
|
return FileResponse(file_path)
|
|
return FileResponse(os.path.join(frontend_dir, "index.html"))
|
|
|
|
|
|
@app.on_event("startup")
|
|
def init_default_owner():
|
|
"""首次启动时创建默认 Owner 账号"""
|
|
from database import SessionLocal
|
|
db = SessionLocal()
|
|
try:
|
|
if not db.query(User).filter(User.role == UserRole.OWNER).first():
|
|
owner = User(
|
|
username="admin",
|
|
password_hash=hash_password("admin123"),
|
|
name="管理员",
|
|
phase_group=PhaseGroup.PRODUCTION,
|
|
role=UserRole.OWNER,
|
|
monthly_salary=0,
|
|
)
|
|
db.add(owner)
|
|
db.commit()
|
|
print("[OK] default admin created: admin / admin123")
|
|
finally:
|
|
db.close()
|