TIANHE
106 lines
3.8 KiB
Python
106 lines
3.8 KiB
Python
"""
|
|
Billing APIs - 会员购买/套餐配置（Mock支付）
|
|
|
|
当前版本先实现“快速商业闭环”的最小可用：
|
|
- 从系统设置(.env)读取 3 档会员（包月/包年/永久）金额与赠送积分配置
|
|
- 用户在前端购买后立即开通/发放积分（后续可替换为真实支付网关）
|
|
"""
|
|
|
|
from flask import Blueprint, jsonify, request, g
|
|
|
|
from app.utils.auth import login_required
|
|
from app.utils.logger import get_logger
|
|
from app.services.billing_service import get_billing_service
|
|
from app.services.usdt_payment_service import get_usdt_payment_service
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
billing_bp = Blueprint("billing", __name__)
|
|
|
|
|
|
@billing_bp.route("/plans", methods=["GET"])
|
|
@login_required
|
|
def get_membership_plans():
|
|
"""Get membership plan configuration + current user's billing snapshot."""
|
|
try:
|
|
user_id = getattr(g, "user_id", None)
|
|
svc = get_billing_service()
|
|
plans = svc.get_membership_plans()
|
|
billing_info = svc.get_user_billing_info(user_id) if user_id else {}
|
|
return jsonify({"code": 1, "msg": "success", "data": {"plans": plans, "billing": billing_info}})
|
|
except Exception as e:
|
|
logger.error(f"get_membership_plans failed: {e}", exc_info=True)
|
|
return jsonify({"code": 0, "msg": str(e), "data": None}), 500
|
|
|
|
|
|
@billing_bp.route("/purchase", methods=["POST"])
|
|
@login_required
|
|
def purchase_membership():
|
|
"""
|
|
Purchase membership (mock: immediate activation).
|
|
|
|
Body:
|
|
{ plan: "monthly" | "yearly" | "lifetime" }
|
|
"""
|
|
try:
|
|
user_id = getattr(g, "user_id", None)
|
|
data = request.get_json() or {}
|
|
plan = (data.get("plan") or "").strip().lower()
|
|
if not plan:
|
|
return jsonify({"code": 0, "msg": "missing_plan", "data": None}), 400
|
|
|
|
success, msg, out = get_billing_service().purchase_membership(user_id, plan)
|
|
if success:
|
|
return jsonify({"code": 1, "msg": msg, "data": out})
|
|
return jsonify({"code": 0, "msg": msg, "data": out}), 400
|
|
except Exception as e:
|
|
logger.error(f"purchase_membership failed: {e}", exc_info=True)
|
|
return jsonify({"code": 0, "msg": str(e), "data": None}), 500
|
|
|
|
|
|
# =========================
|
|
# USDT Pay (方案B)
|
|
# =========================
|
|
|
|
|
|
@billing_bp.route("/usdt/create", methods=["POST"])
|
|
@login_required
|
|
def usdt_create_order():
|
|
"""
|
|
Create USDT order for membership plan (per-order address).
|
|
|
|
Body:
|
|
{ plan: "monthly"|"yearly"|"lifetime" }
|
|
"""
|
|
try:
|
|
user_id = getattr(g, "user_id", None)
|
|
data = request.get_json() or {}
|
|
plan = (data.get("plan") or "").strip().lower()
|
|
if not plan:
|
|
return jsonify({"code": 0, "msg": "missing_plan", "data": None}), 400
|
|
|
|
ok, msg, out = get_usdt_payment_service().create_order(user_id, plan)
|
|
if ok:
|
|
return jsonify({"code": 1, "msg": "success", "data": out})
|
|
return jsonify({"code": 0, "msg": msg, "data": out}), 400
|
|
except Exception as e:
|
|
logger.error(f"usdt_create_order failed: {e}", exc_info=True)
|
|
return jsonify({"code": 0, "msg": str(e), "data": None}), 500
|
|
|
|
|
|
@billing_bp.route("/usdt/order/<int:order_id>", methods=["GET"])
|
|
@login_required
|
|
def usdt_get_order(order_id: int):
|
|
"""Get my USDT order; refresh chain status by default."""
|
|
try:
|
|
user_id = getattr(g, "user_id", None)
|
|
refresh = str(request.args.get("refresh", "1")).lower() in ("1", "true", "yes")
|
|
ok, msg, out = get_usdt_payment_service().get_order(user_id, order_id, refresh=refresh)
|
|
if ok:
|
|
return jsonify({"code": 1, "msg": "success", "data": out})
|
|
return jsonify({"code": 0, "msg": msg, "data": out}), 404
|
|
except Exception as e:
|
|
logger.error(f"usdt_get_order failed: {e}", exc_info=True)
|
|
return jsonify({"code": 0, "msg": str(e), "data": None}), 500
|
|
|