59 lines
2.5 KiB
Python
59 lines
2.5 KiB
Python
"""支付模块路由（/payments/* + /callbacks/*，对应《03-接口文档》第 8 章）"""
|
|
import redis.asyncio as aioredis
|
|
from fastapi import APIRouter, Depends, Header, Request
|
|
from fastapi.responses import JSONResponse, PlainTextResponse
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.core.response import ok
|
|
from app.database import get_db
|
|
from app.deps import Identity, require_user
|
|
from app.redis_client import get_redis
|
|
from app.schemas.payment import RechargeReq
|
|
from app.services import payment_service
|
|
|
|
router = APIRouter(prefix="/payments", tags=["支付"])
|
|
router_callbacks = APIRouter(prefix="/callbacks", tags=["支付回调"])
|
|
|
|
|
|
@router.get("/packages")
|
|
async def list_packages(db: AsyncSession = Depends(get_db), redis: aioredis.Redis = Depends(get_redis),
|
|
identity: Identity = Depends(require_user)):
|
|
return ok(await payment_service.list_packages(db, redis))
|
|
|
|
|
|
@router.post("/recharge")
|
|
async def create_recharge(req: RechargeReq, db: AsyncSession = Depends(get_db),
|
|
redis: aioredis.Redis = Depends(get_redis),
|
|
x_platform: str | None = Header(default=None),
|
|
identity: Identity = Depends(require_user)):
|
|
return ok(await payment_service.create_recharge(db, redis, identity.user_id, req, x_platform))
|
|
|
|
|
|
@router.get("/orders/{order_no}")
|
|
async def get_order(order_no: str, db: AsyncSession = Depends(get_db),
|
|
identity: Identity = Depends(require_user)):
|
|
return ok(await payment_service.get_order(db, identity.user_id, order_no))
|
|
|
|
|
|
@router.post("/orders/{order_no}/cancel")
|
|
async def cancel_order(order_no: str, db: AsyncSession = Depends(get_db),
|
|
identity: Identity = Depends(require_user)):
|
|
await payment_service.cancel_order(db, identity.user_id, order_no)
|
|
return ok()
|
|
|
|
|
|
@router_callbacks.post("/alipay", include_in_schema=False)
|
|
async def alipay_callback(request: Request, db: AsyncSession = Depends(get_db)):
|
|
"""支付宝异步通知：平台要求纯文本 success/failure"""
|
|
form = dict(await request.form())
|
|
result = await payment_service.handle_alipay_callback(db, form)
|
|
return PlainTextResponse(result)
|
|
|
|
|
|
@router_callbacks.post("/wechat", include_in_schema=False)
|
|
async def wechat_callback(request: Request, db: AsyncSession = Depends(get_db)):
|
|
"""微信支付 V3 异步通知：{"code": "SUCCESS"} 或 {"code": "FAIL"}"""
|
|
body = await request.body()
|
|
result = await payment_service.handle_wechat_callback(db, dict(request.headers), body)
|
|
return JSONResponse(result)
|