65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
import os
|
||
import sys
|
||
import logging
|
||
from contextlib import asynccontextmanager
|
||
from fastapi import FastAPI
|
||
from fastapi.staticfiles import StaticFiles
|
||
from fastapi.middleware.cors import CORSMiddleware
|
||
|
||
from app.config import settings, BASE_DIR
|
||
from app.database import init_db
|
||
from app.api import files, extraction, stories, matching, export, settings as settings_api
|
||
|
||
logging.basicConfig(level=logging.INFO)
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
def _get_frontend_dir() -> str:
|
||
"""获取前端静态文件目录（支持 PyInstaller 打包）"""
|
||
if getattr(sys, 'frozen', False):
|
||
# 打包后：exe 同级目录下的 frontend/dist
|
||
return os.path.join(BASE_DIR, "frontend", "dist")
|
||
else:
|
||
# 开发模式：项目根目录下的 frontend/dist
|
||
return os.path.join(os.path.dirname(BASE_DIR), "frontend", "dist")
|
||
|
||
|
||
@asynccontextmanager
|
||
async def lifespan(app: FastAPI):
|
||
"""应用生命周期：启动时初始化数据库"""
|
||
logger.info("初始化数据库...")
|
||
await init_db()
|
||
logger.info("数据库初始化完成")
|
||
yield
|
||
logger.info("应用关闭")
|
||
|
||
|
||
app = FastAPI(
|
||
title="Story Extractor",
|
||
description="直播文字稿故事提取工具",
|
||
version="1.0.0",
|
||
lifespan=lifespan,
|
||
)
|
||
|
||
# CORS
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=["*"],
|
||
allow_credentials=True,
|
||
allow_methods=["*"],
|
||
allow_headers=["*"],
|
||
)
|
||
|
||
# API 路由
|
||
app.include_router(files.router, prefix="/api/v1")
|
||
app.include_router(extraction.router, prefix="/api/v1")
|
||
app.include_router(stories.router, prefix="/api/v1")
|
||
app.include_router(matching.router, prefix="/api/v1")
|
||
app.include_router(export.router, prefix="/api/v1")
|
||
app.include_router(settings_api.router, prefix="/api/v1")
|
||
|
||
# 托管前端静态文件（放在最后，作为 fallback）
|
||
frontend_dist = _get_frontend_dir()
|
||
if os.path.exists(frontend_dist):
|
||
app.mount("/", StaticFiles(directory=frontend_dist, html=True), name="static")
|