from models import db
from datetime import datetime

class RechargeRecord(db.Model):
    __tablename__ = 'recharge_records'
    
    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
    amount = db.Column(db.Numeric(10, 2), nullable=False)  # 美元金额
    actual_amount = db.Column(db.Numeric(10, 2), nullable=False)  # 实际支付金额（人民币）
    payment_method = db.Column(db.String(50), nullable=False)  # alipay, wechat等
    payment_status = db.Column(db.String(20), default='pending')  # pending, completed, failed
    exchange_code = db.Column(db.String(50), nullable=True)  # 兑换码
    transaction_id = db.Column(db.String(100), nullable=True)  # 第三方交易ID
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    completed_at = db.Column(db.DateTime, nullable=True)
    
    def to_dict(self):
        """转换为字典"""
        return {
            'id': self.id,
            'amount': float(self.amount),
            'actual_amount': float(self.actual_amount),
            'payment_method': self.payment_method,
            'payment_status': self.payment_status,
            'created_at': self.created_at.strftime('%Y-%m-%d %H:%M:%S') if self.created_at else None,
            'completed_at': self.completed_at.strftime('%Y-%m-%d %H:%M:%S') if self.completed_at else None,
        }