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.UserModel import UserServiceModel
from app.utils.create_module_service import create_model_service


class ProjectModel(BasicModel, table=True):
  __tablename__ = "pl_project"

  name: str = Field(default=None, description="项目名称")
  description: str = Field(default=None, description="项目描述")
  budget: float = Field(default=None, description="项目预算金额")
  version: int = Field(default=None, description="项目版本号")

  # /*---------------------------------------leader_id-------------------------------------------*/

  leader_id: str = Field(
    default=None,
    description="项目负责人id",
    foreign_key="pl_user.id",  # 添加外键约束
    nullable=True
  )

  leader: Optional["UserServiceModel"] = Relationship(
    sa_relationship_kwargs={
      "foreign_keys": "ProjectModel.leader_id",
      "remote_side": "UserServiceModel.id",
      "uselist": False
    }
  )

  @computed_field
  @property
  def leader_name(self) -> Optional[str]:
    return self.leader.full_name if self.leader else None

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


ProjectService = create_model_service(
  Cls=ProjectModel,
  custom_query=lambda: select(ProjectModel).options(
    selectinload(ProjectModel.leader),
  ),
)