from decimal import Decimal
from typing import Optional, List

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

from app.model.ApproveModel import ApproveModel
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

  # /*---------------------------------------approve_list-------------------------------------------*/

  # approve_list_rel: Optional[List["ApproveModel"]] = Relationship(
  #   sa_relationship_kwargs={
  #     "foreign_keys": "ApproveModel.proj_id",
  #     "primaryjoin": "ProjectModel.id == ApproveModel.proj_id",
  #   }
  # )
  #
  # @computed_field
  # @property
  # def approve_list(self) -> Optional[List["ApproveModel"]]:
  #   return self.approve_list_rel
  #
  # @approve_list.setter
  # def approve_list(self, value: Optional[List["ApproveModel"]]) -> None:
  #   pass
  #
  # # /*---------------------------------------spent-------------------------------------------*/
  # @computed_field
  # @property
  # def spent(self) -> Decimal:
  #   # return sum([approve.amount for approve in self.approve_list_rel if approve.status == 'approved'])
  #   return 0
  #
  # @spent.setter
  # def spent(self, value: Decimal) -> None:
  #   pass


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

    # selectinload(ProjectModel.approve_list_rel),
  ),
)