state.py 633 B

12345678910111213141516171819202122
  1. """每个模型的配色状态读写。"""
  2. from typing import Any
  3. from fastapi import APIRouter, HTTPException, Query
  4. from services import state_service
  5. router = APIRouter()
  6. @router.get("/api/state")
  7. def get_state(model: str = Query(...)) -> dict[str, Any]:
  8. return state_service.load_model_state(model)
  9. @router.post("/api/state")
  10. def save_state(payload: dict[str, Any], model: str = Query(...)) -> dict[str, Any]:
  11. if not model:
  12. raise HTTPException(status_code=400, detail="缺少 model 参数")
  13. state_service.save_model_state(model, payload)
  14. return {"ok": True, "model": model, "count": len(payload)}