34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
|
#!/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()
|