51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
|
"""FastAPI 入口：挂 API + 静态前端。"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import logging
|
|||
|
|
import os
|
|||
|
|
from contextlib import asynccontextmanager
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
from fastapi import FastAPI
|
|||
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|||
|
|
from fastapi.staticfiles import StaticFiles
|
|||
|
|
|
|||
|
|
from .api import router
|
|||
|
|
from .config import get_settings
|
|||
|
|
from .db import init_db
|
|||
|
|
|
|||
|
|
logging.basicConfig(level=logging.INFO, format="%(asctime)s %(levelname)s %(name)s: %(message)s")
|
|||
|
|
log = logging.getLogger("as-event")
|
|||
|
|
settings = get_settings()
|
|||
|
|
|
|||
|
|
|
|||
|
|
@asynccontextmanager
|
|||
|
|
async def lifespan(app: FastAPI):
|
|||
|
|
# 首次启动自动建表（create_all 只补缺表，安全）；生产可用 ddl/own_db_init.sql
|
|||
|
|
if os.getenv("AUTO_CREATE_TABLES", "1") == "1":
|
|||
|
|
try:
|
|||
|
|
init_db()
|
|||
|
|
log.info("自有库建表检查完成")
|
|||
|
|
except Exception as e: # noqa: BLE001
|
|||
|
|
log.warning("建表失败（DB 未就绪？）：%s", e)
|
|||
|
|
yield
|
|||
|
|
|
|||
|
|
|
|||
|
|
app = FastAPI(title="A股大事记录 / 择时看板", version="0.1.0", lifespan=lifespan)
|
|||
|
|
|
|||
|
|
app.add_middleware(
|
|||
|
|
CORSMiddleware,
|
|||
|
|
allow_origins=settings.cors_origin_list,
|
|||
|
|
allow_methods=["*"],
|
|||
|
|
allow_headers=["*"],
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
app.include_router(router)
|
|||
|
|
|
|||
|
|
# 静态前端（放在 API 之后，作为兜底挂在根路径）
|
|||
|
|
_frontend = Path(__file__).resolve().parent.parent / "frontend"
|
|||
|
|
if _frontend.is_dir():
|
|||
|
|
app.mount("/", StaticFiles(directory=str(_frontend), html=True), name="frontend")
|
|||
|
|
else:
|
|||
|
|
log.warning("未找到前端目录: %s", _frontend)
|