#!/usr/bin/env python # -*- coding: utf-8 -*- from typing import Dict, Any, List, Optional from datetime import datetime from fastapi import HTTPException, status from sqlalchemy import Column, Integer, String, DateTime, Float, Index from sqlalchemy.orm import Session from cryptoai.models.base import Base, logger # 定义订阅订单数据模型 class SubscriptionOrder(Base): """订阅订单数据表模型""" __tablename__ = 'subscription_orders' id = Column(Integer, primary_key=True, autoincrement=True) user_id = Column(Integer, nullable=False, comment='用户ID') order_id = Column(String(100), nullable=False, unique=True, comment='订单ID') payment_order_id = Column(String(100), nullable=False, comment='支付订单ID') amount = Column(Float, nullable=False, comment='金额') member_type = Column(Integer, nullable=False, comment='会员类型(1=VIP，2=SVIP)') currency = Column(String(10), nullable=False, comment='货币类型') time_type = Column(Integer, nullable=False, comment='时间类型(1=包月，2=包年)') status = Column(Integer, nullable=False, default=1, comment='订单状态(1=待支付，2=已完成)') create_time = Column(DateTime, nullable=False, default=datetime.now, comment='创建时间') # 索引和表属性 __table_args__ = ( Index('idx_user_id', 'user_id'), Index('idx_order_id', 'order_id'), Index('idx_payment_order_id', 'payment_order_id'), Index('idx_member_type', 'member_type'), Index('idx_status', 'status'), Index('idx_create_time', 'create_time'), {'mysql_charset': 'utf8mb4', 'mysql_collate': 'utf8mb4_unicode_ci'} ) class SubscriptionOrderManager: """订阅订单管理类""" def __init__(self, session: Session = None): self.session = session def create_order(self, user_id: int, order_id: str, payment_order_id: str, amount: float, member_type: int, currency: str, time_type: int, status: int = 1) -> bool: """ 创建订阅订单 Args: user_id: 用户ID order_id: 订单ID payment_order_id: 支付订单ID amount: 金额 member_type: 会员类型(1=VIP，2=SVIP) currency: 货币类型 time_type: 时间类型(1=包月，2=包年) status: 订单状态(1=待支付，2=已完成)，默认为1 Returns: 创建是否成功 """ try: # 验证参数 if member_type not in [1, 2]: logger.warning(f"无效的会员类型: {member_type}") return False if time_type not in [1, 2]: logger.warning(f"无效的时间类型: {time_type}") return False if status not in [1, 2]: logger.warning(f"无效的订单状态: {status}") return False if amount <= 0: logger.warning(f"无效的金额: {amount}") return False # 检查订单ID是否已存在 existing_order = self.session.query(SubscriptionOrder).filter( SubscriptionOrder.order_id == order_id ).first() if existing_order: logger.warning(f"订单ID {order_id} 已存在") return False # 创建新订单 new_order = SubscriptionOrder( user_id=user_id, order_id=order_id, payment_order_id=payment_order_id, amount=amount, member_type=member_type, currency=currency, time_type=time_type, status=status, create_time=datetime.now() ) # 添加并提交 self.session.add(new_order) self.session.commit() logger.info(f"成功创建订阅订单: {order_id}，用户: {user_id}") return True except Exception as e: self.session.rollback() logger.error(f"创建订阅订单失败: {e}") return False def get_order_by_id(self, order_id: str) -> Optional[Dict[str, Any]]: """ 通过订单ID获取订单信息 Args: order_id: 订单ID Returns: 订单信息，如果订单不存在则返回None """ try: # 查询订单 order = self.session.query(SubscriptionOrder).filter( SubscriptionOrder.order_id == order_id ).first() if order: # 转换为字典 return { 'id': order.id, 'user_id': order.user_id, 'order_id': order.order_id, 'payment_order_id': order.payment_order_id, 'amount': order.amount, 'member_type': order.member_type, 'currency': order.currency, 'time_type': order.time_type, 'status': order.status, 'create_time': order.create_time } else: return None except Exception as e: logger.error(f"获取订单信息失败: {e}") return None def get_orders_by_user_id(self, user_id: int) -> List[Dict[str, Any]]: """ 通过用户ID获取用户的所有订单 Args: user_id: 用户ID Returns: 订单列表 """ try: # 查询用户的所有订单 orders = self.session.query(SubscriptionOrder).filter( SubscriptionOrder.user_id == user_id ).order_by(SubscriptionOrder.create_time.desc()).all() # 转换为字典列表 order_list = [] for order in orders: order_list.append({ 'id': order.id, 'user_id': order.user_id, 'order_id': order.order_id, 'payment_order_id': order.payment_order_id, 'amount': order.amount, 'member_type': order.member_type, 'currency': order.currency, 'time_type': order.time_type, 'status': order.status, 'create_time': order.create_time }) return order_list except Exception as e: logger.error(f"获取用户订单列表失败: {e}") return [] def get_order_count(self) -> int: """ 获取订单总数 """ try: # 查询订单数量 order_count = self.session.query(SubscriptionOrder).count() return order_count except Exception as e: logger.error(f"获取订单数量失败: {e}") return 0 def update_order_status(self, order_id: str, status: int) -> bool: """ 更新订单状态 Args: order_id: 订单ID status: 新的订单状态(1=待支付，2=已完成) Returns: 更新是否成功 """ try: # 验证状态值 if status not in [1, 2]: logger.warning(f"无效的订单状态: {status}") return False # 查询订单 order = self.session.query(SubscriptionOrder).filter( SubscriptionOrder.order_id == order_id ).first() if not order: logger.warning(f"订单ID {order_id} 不存在") return False # 更新状态 order.status = status self.session.commit() status_name = "待支付" if status == 1 else "已完成" logger.info(f"成功更新订单 {order_id} 状态为: {status_name}") return True except Exception as e: self.session.rollback() logger.error(f"更新订单状态失败: {e}") return False def get_orders_by_status(self, status: int) -> List[Dict[str, Any]]: """ 根据状态获取订单列表 Args: status: 订单状态(1=待支付，2=已完成) Returns: 订单列表 """ try: # 验证状态值 if status not in [1, 2]: logger.warning(f"无效的订单状态: {status}") return [] # 查询指定状态的订单 orders = self.session.query(SubscriptionOrder).filter( SubscriptionOrder.status == status ).order_by(SubscriptionOrder.create_time.desc()).all() # 转换为字典列表 order_list = [] for order in orders: order_list.append({ 'id': order.id, 'user_id': order.user_id, 'order_id': order.order_id, 'payment_order_id': order.payment_order_id, 'amount': order.amount, 'member_type': order.member_type, 'currency': order.currency, 'time_type': order.time_type, 'status': order.status, 'create_time': order.create_time }) return order_list except Exception as e: logger.error(f"获取订单列表失败: {e}") return []