from enum import Enum
from typing import Optional

from pydantic import computed_field
from sqlalchemy.orm import selectinload
from sqlmodel import Field, Relationship, select

from app.model.BasicModel import BasicModel
from app.model.PosModel import PosModel
from app.utils.create_module_service import create_model_service


class UserValidate(str, Enum):
  Y = 'Y'
  N = 'N'


# 公共的，也是最后返回给前端的一个用户信息数据类型
class PublicUser(BasicModel):
  username: str = Field(..., description="用户名")
  email: str = Field(..., description="邮箱")
  full_name: str = Field(..., description="用户全名")
  valid: UserValidate = Field(default=UserValidate.N, description="用户账号是否已经激活")


# 注册的时候，客户端传入的用户信息，需要包含这个明文密码字段
class RegistryUser(PublicUser):
  password: str


# 对pl_user表进行增删改查时的这个model类
class UserModel(PublicUser, table=True):
  __tablename__ = "pl_user"
  hash_password: str


# 对pl_user表进行增删改查时的这个model类
class UserServiceModel(PublicUser, table=True):
  __tablename__ = "pl_user"
  __table_args__ = {'extend_existing': True}

  pos_code: str = Field(
    default=None,
    description="用户职位编码",
    foreign_key="pl_pos.code",  # 添加外键约束
    nullable=True
  )

  position: Optional["PosModel"] = Relationship(
    sa_relationship_kwargs={
      "foreign_keys": "UserServiceModel.pos_code",
      "remote_side": "PosModel.code",
      "uselist": False
    }
  )

  @computed_field
  @property
  def pos_name(self) -> Optional[str]:
    return self.position.name if self.position else None

  @pos_name.setter
  def pos_name(self, value: Optional[str]) -> None:
    pass


UserService = create_model_service(
  Cls=UserServiceModel,
  custom_query=(lambda: select(UserServiceModel)
                .options(selectinload(UserServiceModel.position)))
)