feat: complete huihui square avatar workflows

digital-avatar-app/backend/routers/tokens.py

@@ -0,0 +1,37 @@
+from fastapi import APIRouter, Depends, Body
+from sqlalchemy.orm import Session
+
+from database import get_db
+from models import TokenAccount, TokenPlan
+from responses import ok, fail
+
+router = APIRouter(tags=["Token"])
+
+
+@router.get("/token/balance")
+def balance(db: Session = Depends(get_db)):
+    acc = db.query(TokenAccount).first()
+    return ok({"balance": acc.balance if acc else 0})
+
+
+@router.get("/token/plans")
+def plans(db: Session = Depends(get_db)):
+    items = db.query(TokenPlan).order_by(TokenPlan.price.asc()).all()
+    return ok([p.to_dict() for p in items])
+
+
+@router.post("/token/charge")
+def charge(payload: dict = Body(...), db: Session = Depends(get_db)):
+    plan_id = payload.get("planId")
+    plan = db.query(TokenPlan).filter(TokenPlan.id == plan_id).first()
+    if not plan:
+        return fail("套餐不存在", 404)
+    acc = db.query(TokenAccount).first()
+    if not acc:
+        acc = TokenAccount(balance=0)
+        db.add(acc)
+        db.commit()
+        db.refresh(acc)
+    acc.balance += plan.amount
+    db.commit()
+    return ok({"balance": acc.balance, "charged": plan.amount})