s''

app/services/payment.py

@@ -0,0 +1,86 @@
+"""支付成功后的统一业务处理（基于 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,
+):
+    """
+    支付成功后的统一业务处理
+    :param use_memos: 是否创建 Memos 用户，仅 nomadvip 为 True
+    """
+    user_id, pay_type = parse_order_id(order_id)
+    if not user_id:
+        return
+
+    print(f"支付成功 [{provider_name}] 用户ID: {user_id}，类型: {pay_type}，订单: {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,
+            })
+        processed_orders.add(order_id)
+    except Exception as e:
+        print(f"PocketBase 操作失败: {e}")
+        logging.error(f"PocketBase 操作失败 - order_id: {order_id}, error: {e}")
+        raise