131 lines
5.6 KiB
Python
131 lines
5.6 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""支付子应用接口层：微信回调 + 算力充值（C 端自服务）。
|
|||
|
|
|
|||
|
|
- ``POST /opc/pay/notify`` 微信支付回调（公网、无登录鉴权，靠 V3 验签）。
|
|||
|
|
- ``/opc/compute/recharge/*`` 用户充值（套餐/下单/状态/记录），require_roles("opc_member")。
|
|||
|
|
"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import json
|
|||
|
|
import logging
|
|||
|
|
|
|||
|
|
from fastapi import APIRouter, Depends, HTTPException, Request, Response
|
|||
|
|
from pydantic import BaseModel, Field
|
|||
|
|
|
|||
|
|
from ..api.dependencies import get_db
|
|||
|
|
from ..infrastructure.repositories import Database
|
|||
|
|
from ..rbac import require_roles
|
|||
|
|
from . import service, wxpay
|
|||
|
|
from .repository import ComputeRechargeOrderRepository
|
|||
|
|
|
|||
|
|
logger = logging.getLogger("pay.api")
|
|||
|
|
|
|||
|
|
router = APIRouter(tags=["pay"])
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _repo(db: Database) -> ComputeRechargeOrderRepository:
|
|||
|
|
"""订单仓储（复用请求级 session；支付域自持仓储，便于整域拆分微服务）。"""
|
|||
|
|
return ComputeRechargeOrderRepository(db.session)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ── 微信支付回调（微信服务器调用，验签解密后到账）─────────────────────────────
|
|||
|
|
@router.post("/opc/pay/notify", summary="微信支付回调（无登录鉴权，V3 验签）")
|
|||
|
|
async def pay_notify(request: Request, db: Database = Depends(get_db)):
|
|||
|
|
body = await request.body()
|
|||
|
|
result = await wxpay.verify_and_decrypt(request.headers, body)
|
|||
|
|
if not result or not isinstance(result, dict):
|
|||
|
|
raise HTTPException(status_code=400, detail="回调验签/解密失败")
|
|||
|
|
# 展平 resource 到顶层（out_trade_no / transaction_id / trade_state ...）
|
|||
|
|
resource = result.pop("resource", {})
|
|||
|
|
if isinstance(resource, dict):
|
|||
|
|
result.update(resource)
|
|||
|
|
order_no = result.get("out_trade_no", "")
|
|||
|
|
logger.info("微信支付回调: order=%s trade_state=%s", order_no, result.get("trade_state"))
|
|||
|
|
try:
|
|||
|
|
ok = await service.handle_notify(db, result)
|
|||
|
|
except Exception as exc: # noqa: BLE001
|
|||
|
|
logger.error("回调处理异常 order=%s: %s", order_no, exc, exc_info=True)
|
|||
|
|
raise HTTPException(status_code=500, detail="处理失败，等待微信重试") from exc
|
|||
|
|
if not ok:
|
|||
|
|
raise HTTPException(status_code=500, detail="处理失败，等待微信重试")
|
|||
|
|
return {"code": "SUCCESS", "message": "成功"}
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ── 用户充值（C 端自服务）─────────────────────────────────────────────────────
|
|||
|
|
class RechargeOrderRequest(BaseModel):
|
|||
|
|
"""充值下单请求：package_id（套餐）或 amount_yuan（自由金额）二选一。"""
|
|||
|
|
client_type: str = Field(default="native", description="native(桌面扫码) | jsapi(小程序)")
|
|||
|
|
package_id: str = ""
|
|||
|
|
amount_yuan: float | None = Field(default=None, ge=0)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.get("/opc/compute/recharge/packages", summary="充值套餐与支付开关")
|
|||
|
|
async def recharge_packages(
|
|||
|
|
db: Database = Depends(get_db),
|
|||
|
|
_u: dict = Depends(require_roles("opc_member")),
|
|||
|
|
):
|
|||
|
|
return await service.packages(db)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.post("/opc/compute/recharge/orders", summary="创建充值订单（native=二维码 / jsapi=小程序支付参数）")
|
|||
|
|
async def recharge_create(
|
|||
|
|
body: RechargeOrderRequest,
|
|||
|
|
db: Database = Depends(get_db),
|
|||
|
|
user: dict = Depends(require_roles("opc_member")),
|
|||
|
|
):
|
|||
|
|
try:
|
|||
|
|
order = await service.create_order(
|
|||
|
|
db, user, client_type=body.client_type,
|
|||
|
|
package_id=body.package_id, amount_yuan=body.amount_yuan,
|
|||
|
|
)
|
|||
|
|
except ValueError as exc:
|
|||
|
|
raise HTTPException(status_code=400, detail=str(exc)) from exc
|
|||
|
|
except RuntimeError as exc:
|
|||
|
|
raise HTTPException(status_code=503, detail=str(exc)) from exc
|
|||
|
|
return order
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.get("/opc/compute/recharge/orders/{order_no}/pay-params", summary="小程序扫码确认支付（归属校验 + JSAPI 参数）")
|
|||
|
|
async def recharge_pay_params(
|
|||
|
|
order_no: str,
|
|||
|
|
db: Database = Depends(get_db),
|
|||
|
|
user: dict = Depends(require_roles("opc_member")),
|
|||
|
|
):
|
|||
|
|
"""微信扫桌面端小程序码 → 打开小程序本接口取 wx.requestPayment 参数。"""
|
|||
|
|
try:
|
|||
|
|
return await service.build_pay_params(db, user, order_no)
|
|||
|
|
except ValueError as exc:
|
|||
|
|
raise HTTPException(status_code=400, detail=str(exc)) from exc
|
|||
|
|
except RuntimeError as exc:
|
|||
|
|
raise HTTPException(status_code=503, detail=str(exc)) from exc
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.post("/opc/compute/recharge/orders/{order_no}/status", summary="查询充值订单状态（含主动对账补单）")
|
|||
|
|
async def recharge_status(
|
|||
|
|
order_no: str,
|
|||
|
|
db: Database = Depends(get_db),
|
|||
|
|
user: dict = Depends(require_roles("opc_member")),
|
|||
|
|
):
|
|||
|
|
repo = _repo(db)
|
|||
|
|
order = await repo.get_by_order_no(order_no)
|
|||
|
|
if order is None or order["user_id"] != user["id"]:
|
|||
|
|
raise HTTPException(status_code=404, detail="订单不存在")
|
|||
|
|
order = await service.reconcile(db, order)
|
|||
|
|
return {
|
|||
|
|
"order_no": order["order_no"], "status": order["status"],
|
|||
|
|
"paid": order["status"] in ("paid", "credited"),
|
|||
|
|
"credited": order["status"] == "credited",
|
|||
|
|
"amount_fen": order["amount_fen"], "quota_micro": order["quota_micro"],
|
|||
|
|
"bonus_quota_micro": order["bonus_quota_micro"],
|
|||
|
|
"transaction_id": order["transaction_id"], "expires_at": order["expires_at"],
|
|||
|
|
"paid_at": order.get("paid_at", ""), "credited_at": order.get("credited_at", ""),
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.get("/opc/compute/recharge/orders", summary="我的充值记录")
|
|||
|
|
async def recharge_orders(
|
|||
|
|
db: Database = Depends(get_db),
|
|||
|
|
user: dict = Depends(require_roles("opc_member")),
|
|||
|
|
):
|
|||
|
|
return {"items": await _repo(db).list_by_user(user["id"])}
|