from typing import Optional

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

from app.model.ProjectModel import ProjectModel
from app.model.PublicApproveModel import PublicApproveModel
from app.utils.create_module_service import create_model_service


# 自动查询project信息的审批单类
class ApproveModel(PublicApproveModel, table=True):
  __tablename__ = "pl_approve"
  __table_args__ = {'extend_existing': True}

  # /*---------------------------------------project-------------------------------------------*/

  proj_id: str = Field(
    default=None,
    description="所属项目id",
    foreign_key="pl_project.id",  # 添加外键约束
    nullable=True
  )

  proj_rel: Optional["ProjectModel"] = Relationship(
    sa_relationship_kwargs={
      "foreign_keys": "ApproveModel.proj_id",
      "remote_side": "ProjectModel.id",
      "uselist": False
    }
  )

  @computed_field
  @property
  def project(self) -> Optional[ProjectModel]:
    return self.proj_rel

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


ApproveProjService = create_model_service(
  Cls=ApproveModel,
  custom_query=lambda: select(ApproveModel)
  .options(
    selectinload(ApproveModel.proj_rel).
    selectinload(ProjectModel.leader)
  )
)