Joey
70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
from pathlib import Path
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.staticfiles import StaticFiles
|
|
from sqlalchemy import text
|
|
|
|
from api import ai, auth, categories, collection, fx, image, ozon, products, publish, shops
|
|
from config import get_settings
|
|
from db import get_engine
|
|
|
|
# web/ 是 v1 工具台，留在仓库根，故上跳一级
|
|
WEB_DIR = Path(__file__).resolve().parents[1] / "web"
|
|
# 本地存储（开发兜底）媒体目录
|
|
MEDIA_DIR = Path(__file__).resolve().parents[1] / "data" / "media"
|
|
|
|
app = FastAPI(title="Ozon Seller Kit", version="0.2.0")
|
|
|
|
settings = get_settings()
|
|
if settings.cors_origin_list:
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=settings.cors_origin_list,
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# 业务路由
|
|
app.include_router(auth.router)
|
|
app.include_router(collection.router)
|
|
app.include_router(products.router)
|
|
app.include_router(shops.router)
|
|
app.include_router(categories.router)
|
|
app.include_router(publish.router)
|
|
app.include_router(fx.router)
|
|
app.include_router(ai.router)
|
|
app.include_router(image.router)
|
|
app.include_router(ozon.router)
|
|
|
|
|
|
@app.on_event("startup")
|
|
async def on_startup() -> None:
|
|
# 开发便利：确保表存在（生产以 Alembic 迁移为准，create_all 幂等不删表）
|
|
from db import Base
|
|
import models # noqa: F401
|
|
|
|
MEDIA_DIR.mkdir(parents=True, exist_ok=True)
|
|
async with get_engine().begin() as conn:
|
|
await conn.run_sync(Base.metadata.create_all)
|
|
|
|
|
|
@app.get("/api/health")
|
|
async def health() -> dict:
|
|
db_ok = True
|
|
try:
|
|
async with get_engine().connect() as conn:
|
|
await conn.execute(text("SELECT 1"))
|
|
except Exception: # noqa: BLE001
|
|
db_ok = False
|
|
return {"status": "ok" if db_ok else "degraded", "db": db_ok}
|
|
|
|
|
|
# 本地媒体（开发兜底存储）
|
|
MEDIA_DIR.mkdir(parents=True, exist_ok=True)
|
|
app.mount("/media", StaticFiles(directory=str(MEDIA_DIR)), name="media")
|
|
|
|
if WEB_DIR.is_dir():
|
|
app.mount("/", StaticFiles(directory=str(WEB_DIR), html=True), name="web")
|