#!/usr/bin/env python3
"""测试 VIP 功能"""
from datetime import datetime
from lover.db import SessionLocal
from lover.models import User

def test_vip_status():
    db = SessionLocal()
    try:
        # 测试用户 70 和 84
        for user_id in [70, 84]:
            user = db.query(User).filter(User.id == user_id).first()
            if not user:
                print(f"用户 {user_id} 不存在")
                continue
            
            current_timestamp = int(datetime.utcnow().timestamp())
            is_vip = user.vip_endtime and user.vip_endtime > current_timestamp
            
            print(f"\n用户 {user_id} ({user.nickname}):")
            print(f"  VIP 到期时间戳: {user.vip_endtime}")
            if user.vip_endtime:
                vip_end_date = datetime.fromtimestamp(user.vip_endtime)
                print(f"  VIP 到期日期: {vip_end_date}")
            print(f"  当前时间戳: {current_timestamp}")
            print(f"  是否 VIP: {is_vip}")
            print(f"  视频生成次数: {user.video_gen_remaining}")
            print(f"  上次重置日期: {user.video_gen_reset_date}")
    finally:
        db.close()

if __name__ == "__main__":
    test_vip_status()