90 lines
2.8 KiB
Python
90 lines
2.8 KiB
Python
"""FastAPI 主应用 - with SPA static file serving"""
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.staticfiles import StaticFiles
|
|
from fastapi.responses import FileResponse
|
|
from pathlib import Path
|
|
from contextlib import asynccontextmanager
|
|
|
|
from .config import settings
|
|
from .database import Base, engine
|
|
from .api.v1 import jobs, upload, results, tasks
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
"""应用生命周期管理"""
|
|
# 启动时
|
|
print("🚀 Starting BtToxin Pipeline API...")
|
|
# 创建数据库表
|
|
Base.metadata.create_all(bind=engine)
|
|
print("✅ Database tables created")
|
|
yield
|
|
# 关闭时
|
|
print("👋 Shutting down BtToxin Pipeline API...")
|
|
|
|
|
|
app = FastAPI(
|
|
title=settings.APP_NAME,
|
|
version=settings.APP_VERSION,
|
|
description="Automated Bacillus thuringiensis toxin mining pipeline",
|
|
lifespan=lifespan,
|
|
docs_url="/api/docs",
|
|
redoc_url="/api/redoc",
|
|
openapi_url="/api/openapi.json"
|
|
)
|
|
|
|
# CORS
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=settings.CORS_ORIGINS,
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# 路由
|
|
app.include_router(jobs.router, prefix=f"{settings.API_V1_STR}/jobs", tags=["jobs"])
|
|
app.include_router(tasks.router, prefix=f"{settings.API_V1_STR}/tasks", tags=["tasks"])
|
|
app.include_router(upload.router, prefix=f"{settings.API_V1_STR}/upload", tags=["upload"])
|
|
app.include_router(results.router, prefix=f"{settings.API_V1_STR}/results", tags=["results"])
|
|
|
|
|
|
@app.get("/api/meta")
|
|
async def root_meta():
|
|
return {
|
|
"name": settings.APP_NAME,
|
|
"version": settings.APP_VERSION,
|
|
"status": "healthy"
|
|
}
|
|
|
|
|
|
@app.get("/health")
|
|
async def health():
|
|
return {"status": "ok"}
|
|
|
|
|
|
# Mount static files for frontend (in production)
|
|
# The frontend dist files are served from /app/frontend/dist in the container
|
|
frontend_dist_path = Path("/app/frontend/dist")
|
|
if frontend_dist_path.exists():
|
|
app.mount("/assets", StaticFiles(directory=str(frontend_dist_path / "assets")), name="assets")
|
|
|
|
@app.get("/{full_path:path}", include_in_schema=False)
|
|
async def serve_spa(full_path: str):
|
|
"""Serve frontend SPA - all routes go to index.html"""
|
|
# 如果是 API 路径，且没有被前面路由捕获，返回 404
|
|
if full_path.startswith("api/"):
|
|
return {"detail": "Not Found"}, 404
|
|
|
|
# 处理根路径 ""
|
|
if full_path == "" or full_path == "/":
|
|
return FileResponse(frontend_dist_path / "index.html")
|
|
|
|
file_path = frontend_dist_path / full_path
|
|
if file_path.exists() and file_path.is_file():
|
|
return FileResponse(file_path)
|
|
|
|
# SPA 路由：返回 index.html
|
|
return FileResponse(frontend_dist_path / "index.html")
|