From 1aa6f778b1f140162e0164315632ab797394aca3 Mon Sep 17 00:00:00 2001 From: martsforever Date: Sat, 23 Aug 2025 09:45:17 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20pos=20model=E5=85=B3=E8=81=94=E6=9F=A5?= =?UTF-8?q?=E8=AF=A2=E7=88=B6=E8=81=8C=E4=BD=8D=E4=BF=A1=E6=81=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/model/PosModel.py | 31 +++++++++++++++++++++++++++++-- 1 file changed, 29 insertions(+), 2 deletions(-) diff --git a/app/model/PosModel.py b/app/model/PosModel.py index 973ee88..0446b40 100644 --- a/app/model/PosModel.py +++ b/app/model/PosModel.py @@ -1,4 +1,7 @@ -from sqlmodel import Field +from typing import Optional + +from pydantic import computed_field +from sqlmodel import Field, Relationship from app.model.BasicModel import BasicModel from app.utils.create_module_service import create_model_service @@ -9,9 +12,33 @@ class PosModel(BasicModel, table=True): name: str = Field(default=None, description="职位名称") code: str = Field(default=None, description="职位编码") - parent_code: str = Field(default=None, description="父职位编码") + # parent_code: str = Field(default=None, description="父职位编码") organization_code: str = Field(default=None, description="所属组织编码") remarks: str = Field(default=None, description="备注信息") + parent_code: str = Field( + default=None, + description="职位编码", + foreign_key="pl_pos.code", # 添加外键约束 + nullable=True + ) + + parent: Optional["PosModel"] = Relationship( + sa_relationship_kwargs={ + "foreign_keys": "PosModel.parent_code", + "remote_side": "PosModel.code", + "uselist": False + } + ) + + @computed_field + @property + def parent_name(self) -> Optional[str]: + return self.parent.name if self.parent else None + + @parent_name.setter + def parent_name(self, value: Optional[str]) -> None: + pass + PosService = create_model_service(PosModel)