86 lines
2.4 KiB
Python
86 lines
2.4 KiB
Python
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.OrgModel import OrgModel
|
|
from app.utils.create_module_service import create_model_service
|
|
|
|
|
|
class PosModel(BasicModel, table=True):
|
|
__tablename__ = "pl_pos"
|
|
|
|
name: str = Field(default=None, description="职位名称")
|
|
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-------------------------------------------*/
|
|
|
|
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]:
|
|
try:
|
|
return self.parent.name if self.parent else None
|
|
except:
|
|
return None
|
|
|
|
@parent_name.setter
|
|
def parent_name(self, value: Optional[str]) -> None:
|
|
pass
|
|
|
|
# /*---------------------------------------organization_code-------------------------------------------*/
|
|
|
|
organization_code: str = Field(
|
|
default=None,
|
|
description="所属组织编码",
|
|
foreign_key="pl_org.code", # 添加外键约束
|
|
nullable=True
|
|
)
|
|
|
|
organization: Optional["OrgModel"] = Relationship(
|
|
sa_relationship_kwargs={
|
|
"foreign_keys": "PosModel.organization_code",
|
|
"remote_side": "OrgModel.code",
|
|
"uselist": False
|
|
}
|
|
)
|
|
|
|
@computed_field
|
|
@property
|
|
def organization_name(self) -> Optional[str]:
|
|
try:
|
|
return self.organization.name if self.organization else None
|
|
except:
|
|
return None
|
|
|
|
@organization_name.setter
|
|
def organization_name(self, value: Optional[str]) -> None:
|
|
pass
|
|
|
|
|
|
PosService = create_model_service(
|
|
Cls=PosModel,
|
|
custom_query=(lambda: select(PosModel)
|
|
.options(selectinload(PosModel.parent))
|
|
.options(selectinload(PosModel.organization))),
|
|
)
|