TradeMate Dev
d2736d1ef6
- AI routing rules now stored in system_configs DB table instead of hardcoded config - Multi-model support via name|model composite key for same-provider routing - UnifiedPayService with HMAC-SHA256 gateway integration (alipay/wechat) - Admin payment panel: list, stats, search, filter, refund - WeChat mini-program CI/CD via miniprogram-ci (v1.0.9) - Translation quota extended to LLM provider tier - SearchService with DB-driven provider config (bing/google_cse/searxng) - Footer cleanup across admin/workspace/uni-app - Private key excluded from git tracking
147 lines
4.1 KiB
Python
147 lines
4.1 KiB
Python
from fastapi import APIRouter, Depends, HTTPException, Request, Query
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from pydantic import BaseModel
|
|
from typing import Optional
|
|
from app.database import get_db
|
|
from app.services.payment import PaymentService
|
|
from app.services.unified_pay import UnifiedPayService
|
|
from app.api.v1.deps import get_current_user_id
|
|
import logging
|
|
logger = logging.getLogger(__name__)
|
|
router = APIRouter()
|
|
|
|
|
|
class CreateOrderRequest(BaseModel):
|
|
plan: str
|
|
pay_type: str = "alipay"
|
|
|
|
|
|
class RefundRequest(BaseModel):
|
|
order_no: str
|
|
reason: str = ""
|
|
|
|
|
|
@router.get("/plans")
|
|
async def get_plans():
|
|
svc = PaymentService(None)
|
|
return await svc.get_plans()
|
|
|
|
|
|
@router.get("/subscription")
|
|
async def get_subscription(
|
|
user_id: str = Depends(get_current_user_id),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
svc = PaymentService(db)
|
|
return await svc.get_current_subscription(user_id)
|
|
|
|
|
|
@router.post("/create-order")
|
|
async def create_order(
|
|
data: CreateOrderRequest,
|
|
user_id: str = Depends(get_current_user_id),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
svc = PaymentService(db)
|
|
try:
|
|
return await svc.create_order(user_id, data.plan, data.pay_type)
|
|
except ValueError as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
|
|
|
|
@router.get("/query/{order_no}")
|
|
async def query_payment(
|
|
order_no: str,
|
|
user_id: str = Depends(get_current_user_id),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
svc = PaymentService(db)
|
|
try:
|
|
return await svc.query_payment(user_id, order_no)
|
|
except ValueError as e:
|
|
raise HTTPException(status_code=404, detail=str(e))
|
|
|
|
|
|
@router.get("/transactions")
|
|
async def list_transactions(
|
|
page: int = Query(1, ge=1),
|
|
size: int = Query(20, ge=1, le=100),
|
|
user_id: str = Depends(get_current_user_id),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
svc = PaymentService(db)
|
|
return await svc.list_transactions(user_id, page, size)
|
|
|
|
|
|
@router.post("/refund")
|
|
async def refund(
|
|
data: RefundRequest,
|
|
user_id: str = Depends(get_current_user_id),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
svc = PaymentService(db)
|
|
try:
|
|
return await svc.refund(user_id, data.order_no, data.reason)
|
|
except ValueError as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
|
|
|
|
@router.post("/close-order")
|
|
async def close_order(
|
|
data: dict,
|
|
user_id: str = Depends(get_current_user_id),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
order_no = data.get("order_no", "")
|
|
svc = PaymentService(db)
|
|
try:
|
|
return await svc.close_order(user_id, order_no)
|
|
except ValueError as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
|
|
|
|
@router.get("/query-refund/{order_no}")
|
|
async def query_refund(
|
|
order_no: str,
|
|
user_id: str = Depends(get_current_user_id),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
svc = PaymentService(db)
|
|
try:
|
|
return await svc.query_refund(order_no, user_id=user_id)
|
|
except ValueError as e:
|
|
raise HTTPException(status_code=404, detail=str(e))
|
|
|
|
|
|
@router.post("/webhook")
|
|
async def unified_webhook(request: Request, db: AsyncSession = Depends(get_db)):
|
|
body = await request.body()
|
|
body_str = body.decode("utf-8")
|
|
|
|
gw = UnifiedPayService()
|
|
if not gw.verify_callback(dict(request.headers), body_str):
|
|
logger.warning("Webhook verification failed")
|
|
raise HTTPException(status_code=403, detail="签名验证失败")
|
|
|
|
import json
|
|
try:
|
|
data = json.loads(body_str)
|
|
except json.JSONDecodeError:
|
|
raise HTTPException(status_code=400, detail="无效的 JSON")
|
|
|
|
event = data.get("event", "")
|
|
pay_data = data.get("data", {})
|
|
merchant_order_id = pay_data.get("merchant_order_id", "")
|
|
order_id = pay_data.get("order_id", "")
|
|
transaction_id = pay_data.get("transaction_id", "")
|
|
amount = pay_data.get("amount", 0)
|
|
success = event in ("recharge.completed", "order.refunded")
|
|
|
|
svc = PaymentService(db)
|
|
await svc.handle_callback(
|
|
merchant_order_id, order_id, transaction_id,
|
|
success if event == "recharge.completed" else True,
|
|
amount, body_str,
|
|
)
|
|
return {"code": 0, "message": "OK"}
|