from flask import Blueprint, request, redirect, url_for, session, jsonify, render_template from extensions import db from models import Order, User, to_bj_time, get_bj_now from services.alipay_service import AlipayService from services.logger import system_logger import uuid from datetime import datetime, timedelta payment_bp = Blueprint('payment', __name__, url_prefix='/payment') # 积分价格配置 POINTS_PACKAGES = { '50': {'points': 50, 'amount': 5.00}, '200': {'points': 200, 'amount': 20.00}, '1000': {'points': 1000, 'amount': 100.00}, '5000': {'points': 5000, 'amount': 500.00}, } @payment_bp.route('/create', methods=['POST']) def create_payment(): if 'user_id' not in session: return jsonify({'code': 401, 'msg': '请先登录'}), 401 package_id = request.form.get('package_id') if package_id not in POINTS_PACKAGES: return jsonify({'code': 400, 'msg': '无效的套餐'}), 400 package = POINTS_PACKAGES[package_id] user_id = session['user_id'] # 生成唯一订单号 (时间戳 + 随机位) out_trade_no = datetime.now().strftime('%Y%m%d%H%M%S') + str(uuid.uuid4().hex[:6]) # 创建订单记录 try: order = Order( out_trade_no=out_trade_no, user_id=user_id, amount=package['amount'], points=package['points'], status='PENDING' ) db.session.add(order) db.session.commit() system_logger.info(f"用户创建充值订单", order_id=out_trade_no, amount=package['amount'], points=package['points']) except Exception as e: db.session.rollback() system_logger.error(f"订单创建失败: {str(e)}") return f"订单创建失败: {str(e)}", 500 # 获取支付链接 try: alipay_service = AlipayService() pay_url = alipay_service.create_order_url( out_trade_no=out_trade_no, total_amount=package['amount'], subject=f"购买{package['points']}积分" ) return redirect(pay_url) except Exception as e: system_logger.error(f"支付链接生成失败: {str(e)}") return f"支付链接生成失败: {str(e)}", 500 @payment_bp.route('/return') def payment_return(): """支付成功后的同步跳转页面""" try: data = request.args.to_dict() signature = data.get("sign") if not signature: return "参数错误：缺少签名", 400 alipay_service = AlipayService() success = alipay_service.verify_notify(data, signature) out_trade_no = data.get('out_trade_no') if success: return redirect(url_for('auth.buy_page', success='true', out_trade_no=out_trade_no)) else: system_logger.warning(f"支付同步回调验证失败", order_id=out_trade_no) return "支付验证失败", 400 except Exception as e: system_logger.error(f"处理同步回调异常: {str(e)}") return f"处理支付回调失败: {str(e)}", 500 @payment_bp.route('/history', methods=['GET']) def payment_history(): """获取当前用户的充值历史记录""" if 'user_id' not in session: return redirect(url_for('auth.login')) thirty_min_ago = get_bj_now() - timedelta(minutes=30) user_id = session['user_id'] orders = Order.query.filter( Order.user_id == user_id, db.or_( Order.status == 'PAID', db.and_(Order.status == 'PENDING', Order.created_at >= thirty_min_ago) ) ).order_by(Order.created_at.desc()).all() return render_template('recharge_history.html', orders=orders, modules={'datetime': datetime}) @payment_bp.route('/api/history', methods=['GET']) def api_payment_history(): """API 获取当前用户的充值历史记录""" if 'user_id' not in session: return jsonify({'code': 401, 'msg': '请先登录'}), 401 thirty_min_ago = get_bj_now() - timedelta(minutes=30) user_id = session['user_id'] orders = Order.query.filter( Order.user_id == user_id, db.or_( Order.status == 'PAID', db.and_(Order.status == 'PENDING', Order.created_at >= thirty_min_ago) ) ).order_by(Order.created_at.desc()).all() return jsonify({ "orders": [{ "id": o.id, "out_trade_no": o.out_trade_no, "amount": float(o.amount), "points": o.points, "status": o.status, "trade_no": o.trade_no, "created_at": o.created_at_bj.strftime('%Y-%m-%d %H:%M:%S'), "paid_at": o.paid_at_bj.strftime('%Y-%m-%d %H:%M:%S') if o.paid_at else None } for o in orders] }) @payment_bp.route('/notify', methods=['POST']) def payment_notify(): """支付宝异步通知""" try: data = request.form.to_dict() signature = data.get("sign") if not signature: return "fail" alipay_service = AlipayService() success = alipay_service.verify_notify(data, signature) if success and data.get('trade_status') in ['TRADE_SUCCESS', 'TRADE_FINISHED']: out_trade_no = data.get('out_trade_no') trade_no = data.get('trade_no') order = Order.query.filter_by(out_trade_no=out_trade_no).first() if order and order.status == 'PENDING': order.status = 'PAID' order.trade_no = trade_no order.paid_at = get_bj_now() user = db.session.get(User, order.user_id) if user: user.points += order.points system_logger.info(f"订单支付成功", order_id=out_trade_no, points=order.points, user_id=user.id) # ========== 邀请奖励逻辑 ========== if user.invited_by: from models import InviteReward # 统计该被邀请人之前已经完成多少次充值（不含本次） paid_count = Order.query.filter( Order.user_id == user.id, Order.status == 'PAID', Order.id != order.id # 排除本次订单 ).count() # 只有前3次充值才有奖励（paid_count 是之前的次数，0/1/2 表示这是第1/2/3次） if paid_count < 3: inviter = db.session.get(User, user.invited_by) if inviter: # 计算10%奖励积分（四舍五入） reward_points = round(order.points * 0.1) if reward_points > 0: inviter.points += reward_points # 记录邀请奖励 invite_reward = InviteReward( inviter_id=inviter.id, invitee_id=user.id, order_id=order.id, reward_points=reward_points, recharge_count=paid_count + 1 # 第几次充值 ) db.session.add(invite_reward) system_logger.info( f"邀请奖励发放成功", inviter_id=inviter.id, invitee_id=user.id, reward_points=reward_points, recharge_count=paid_count + 1 ) # ========== 邀请奖励逻辑结束 ========== db.session.commit() return "success" elif order: return "success" else: return "fail" else: return "fail" except Exception as e: system_logger.error(f"处理异步通知异常: {str(e)}") db.session.rollback() return "fail"