import enum
from datetime import datetime
from sqlalchemy import Boolean, Column, Integer, String, DateTime, Enum
from .device import Base


class UserRoleEnum(str, enum.Enum):
    admin = "admin"       # 管理员：可管理设备
    viewer = "viewer"     # 查看者：仅查看数据


class User(Base):
    """本地用户表，关联 Casdoor"""
    __tablename__ = "users"

    id = Column(Integer, primary_key=True, autoincrement=True)
    casdoor_uid = Column(String(128), unique=True, nullable=False, comment="Casdoor 中的用户 ID")
    username = Column(String(128), nullable=False, comment="用户名")
    display_name = Column(String(128), default="", comment="显示名称")
    role = Column(Enum(UserRoleEnum), nullable=False, default=UserRoleEnum.viewer, comment="角色")
    wecom_userid = Column(String(128), default="", comment="企业微信 UserID（用于个人通知）")
    is_active = Column(Boolean, default=True, comment="是否启用")
    last_login_at = Column(DateTime, nullable=True, comment="最后登录时间")
    created_at = Column(DateTime, default=datetime.now, comment="创建时间")

    def __repr__(self):
        return f"<User(id={self.id}, name={self.username}, role={self.role})>"