Files
DP/Server/app/models/user.py

52 lines
2.3 KiB
Python

from sqlalchemy import Column, Integer, String, DateTime, func, ForeignKey, Text, Boolean, Date
from sqlalchemy.orm import relationship
from app.db import Base
class User(Base):
# 用户表定义
__tablename__ = "users"
id = Column(Integer, primary_key=True, index=True)
username = Column(String(64), unique=True, index=True, nullable=False) # 用户名唯一
hashed_password = Column(String(128), nullable=False) # 加密后的密码
created_at = Column(DateTime(timezone=True), server_default=func.now(), nullable=False) # 创建时间
group_id = Column(Integer, ForeignKey("plugin_groups.id"), nullable=True)
permissions = Column(Text, nullable=True) # Comma separated permissions
expire_date = Column(DateTime(timezone=True), nullable=True)
# Email & Verification
email = Column(String(255), unique=True, index=True, nullable=True)
is_verified = Column(Boolean, default=False)
verification_code = Column(String(6), nullable=True)
# Password Reset
reset_token = Column(String(128), nullable=True)
reset_token_expire = Column(DateTime(timezone=True), nullable=True)
# Profile & VIP & Check-in
nickname = Column(String(50), nullable=True)
avatar = Column(String(500), nullable=True)
points = Column(Integer, default=0)
level = Column(Integer, default=1)
vip_type = Column(String(20), default='none') # 'none', 'vip', 'svip'
vip_expire = Column(DateTime(timezone=True), nullable=True)
vip_daily_quota = Column(Integer, default=0)
vip_quota_reset_date = Column(Date, nullable=True)
# User-owned AI credentials (backend forwards requests using the user's own key)
ai_provider = Column(String(32), nullable=True)
ai_api_key = Column(Text, nullable=True)
ai_base_url = Column(String(255), nullable=True)
ai_chat_base_url = Column(String(255), nullable=True)
ai_vision_base_url = Column(String(255), nullable=True)
ai_image_base_url = Column(String(255), nullable=True)
ai_model = Column(String(120), nullable=True)
ai_vision_model = Column(String(120), nullable=True)
ai_image_model = Column(String(120), nullable=True)
total_check_in_days = Column(Integer, default=0)
consecutive_check_in = Column(Integer, default=0)
last_check_in_date = Column(Date, nullable=True)
group = relationship("PluginGroup", back_populates="users")