feat: ProjectModel

app/model/ProjectModel.py

@@ -0,0 +1,49 @@
+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="项目预算金额")
+  spent: float = Field(default=None, description="项目已花费金额")
+  version: int = Field(default=None, description="项目版本号")
+
+  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)),
+)