from Modules.AdminUser.UserUtils import UserUtils

from Utils.ObjUtil import SpecObject
from Utils.ValidateUtil import ValidateAttr, Validate


class User(SpecObject):
    """管理端用户"""

    uid = ValidateAttr(field="uid", type=str)
    email = ValidateAttr(field="email", func=Validate.email)
    name = ValidateAttr(field="name", type=str)
    pwd = ValidateAttr(field="pwd", func=Validate.password)
    status = ValidateAttr(field="status", in_list=UserUtils.user_enum("User", "status"))
    role = ValidateAttr(field="role", in_list=UserUtils.user_enum("User", "role"))
    create_time = ValidateAttr(field="create_time", func=Validate.time_format)

    fields_map = {
        "uid": "用户ID",
        "email": "邮箱",
        "name": "用户名",
        "pwd": "密码",
        "status": "用户状态",
        "role": "角色",
        "create_time": "创建时间"
    }


class EmailVcodeChecker(SpecObject):
    """邮件验证码对象"""

    email = ValidateAttr(field="email", func=Validate.email)
    vcode = ValidateAttr(field="vcode", type=str, length=6)

    fields_map = {
        "email": "邮箱",
        "vcode": "验证码"
    }