from enum import unique, Enum

from pydantic import BaseModel

from typing import Union


class LoginToAppReq(BaseModel):
    code: str


class LoginToAppRes(BaseModel):
    registered: bool
    token: str


class GetUserInfoReq(BaseModel):
    code: str


class GetUserInfoRes(BaseModel):
    openid: str
    email: str
    post: str
    name: str
    department: str
    registered: bool


class TokenData(BaseModel):
    openid: str
    registered: bool
    auth_data: Union[str, None]
    department: Union[str, None]
    post: Union[str, None]
    email: Union[str, None]


class UserInfoChange(BaseModel):
    openid: str
    email: str
    name: str
    post: str
    department: str
    registered: bool


class UserInfo(BaseModel):
    openid: str
    email: str
    post: str
    department: str
    registered: bool


class ChangeUserInfoReq(UserInfoChange):
    email_code: str


class BindEmailReq(BaseModel):
    email: str = "xxxx@fecr.com.cn"
    email_code: str


class EmailSendReqBody(BaseModel):
    email: str = "xxxx@fecr.com.cn"


@unique
class DepartmentTypeEnum(Enum):
    enum01 = "董监高"
    enum02 = "评级部门"
    enum03 = "业务部门"
    enum04 = "研发部门"
    enum05 = "职能部门"
    enum06 = "战略部门"
    enum07 = "子分公司"
    enum08 = "质量部门"