115 lines
4.1 KiB
Python
115 lines
4.1 KiB
Python
"""支付成功后的统一业务处理（基于 SDK）"""
|
||
import logging
|
||
|
||
from .pb import get_pb_client
|
||
from .memos import create_memos_user
|
||
|
||
# 已处理订单（防重复）
|
||
processed_orders: set[str] = set()
|
||
|
||
|
||
def parse_order_id(order_id: str) -> tuple[str, str]:
|
||
"""从 order_id 解析 user_id 和 pay_type"""
|
||
if "_order_" not in order_id:
|
||
return "", "unknown"
|
||
prefix = order_id.split("_order_")[0]
|
||
if "_" in prefix:
|
||
parts = prefix.rsplit("_", 1)
|
||
return parts[0], parts[1].lower()
|
||
return prefix, "unknown"
|
||
|
||
|
||
def handle_payment_success(
|
||
order_id: str,
|
||
pay_price: str,
|
||
provider_name: str,
|
||
*,
|
||
use_memos: bool = False,
|
||
site_id: str | None = None,
|
||
):
|
||
"""
|
||
支付成功后的统一业务处理
|
||
:param use_memos: 是否创建 Memos 用户，仅 nomadvip 为 True
|
||
:param site_id: 站点标识（digital/meetup/vip），用于写入 site_vip
|
||
"""
|
||
user_id, pay_type = parse_order_id(order_id)
|
||
if not user_id:
|
||
return
|
||
|
||
print(f"支付成功 [{provider_name}] 用户ID: {user_id}，类型: {pay_type}，站点: {site_id}，订单: {order_id}")
|
||
|
||
if pay_type == "vip" and use_memos:
|
||
create_memos_user(user_id)
|
||
|
||
if pay_type == "meetup":
|
||
logging.info(f"meetup 类型支付成功，不创建 memos 用户 - user_id: {user_id}")
|
||
|
||
valid_types = ["vip", "book", "meetup", "video"]
|
||
if pay_type not in valid_types:
|
||
pay_type = "vip"
|
||
|
||
amount_cents = int(float(pay_price or 0) * 100)
|
||
|
||
try:
|
||
pb_client = get_pb_client()
|
||
if pay_type == "meetup":
|
||
solan = pb_client.collection("solan")
|
||
existing = solan.get_list(1, 1, {
|
||
"filter": f'user_id = "{user_id}"',
|
||
"sort": "-created",
|
||
})
|
||
if existing.items:
|
||
record_id = existing.items[0].id
|
||
solan.update(record_id, {
|
||
"type": "meetup",
|
||
"order_id": order_id,
|
||
"amount": amount_cents,
|
||
})
|
||
logging.info(f"meetup 支付字段补齐成功 - user_id: {user_id}, record_id: {record_id}")
|
||
else:
|
||
solan.create({
|
||
"user_id": user_id,
|
||
"type": "meetup",
|
||
"order_id": order_id,
|
||
"amount": amount_cents,
|
||
})
|
||
logging.info(f"meetup 完整记录创建（兜底） - user_id: {user_id}")
|
||
else:
|
||
pb_client.collection("payments").create({
|
||
"user_id": user_id,
|
||
"type": pay_type,
|
||
"order_id": order_id,
|
||
"amount": amount_cents,
|
||
})
|
||
|
||
# 写入 site_vip（站点独立 VIP）。meetup 类型统一写 site_id=meetup
|
||
effective_site_id = "meetup" if pay_type == "meetup" else site_id
|
||
if effective_site_id:
|
||
try:
|
||
site_vip = pb_client.collection("site_vip")
|
||
existing_vip = site_vip.get_list(1, 1, {
|
||
"filter": f'user_id = "{user_id}" && site_id = "{effective_site_id}"',
|
||
})
|
||
if existing_vip.items:
|
||
site_vip.update(existing_vip.items[0].id, {
|
||
"order_id": order_id,
|
||
"expires_at": "",
|
||
})
|
||
logging.info(f"site_vip 更新成功 - user_id: {user_id}, site_id: {effective_site_id}")
|
||
else:
|
||
site_vip.create({
|
||
"user_id": user_id,
|
||
"site_id": effective_site_id,
|
||
"order_id": order_id,
|
||
"expires_at": "",
|
||
})
|
||
logging.info(f"site_vip 创建成功 - user_id: {user_id}, site_id: {effective_site_id}")
|
||
except Exception as e:
|
||
logging.warning(f"site_vip 写入失败（不影响主流程）: {e}")
|
||
|
||
processed_orders.add(order_id)
|
||
except Exception as e:
|
||
print(f"PocketBase 操作失败: {e}")
|
||
logging.error(f"PocketBase 操作失败 - order_id: {order_id}, error: {e}")
|
||
raise
|