feat: 关联查询出来报销单的差旅费用子数据

This commit is contained in:
martsforever
2025-08-23 17:28:36 +08:00
parent 815047208a
commit bdbfb95d73
+27 -2
View File
@@ -1,4 +1,4 @@
from typing import Optional from typing import Optional, List
from pydantic import computed_field from pydantic import computed_field
from sqlalchemy.orm import selectinload from sqlalchemy.orm import selectinload
@@ -6,6 +6,7 @@ from sqlmodel import Field, Relationship, select
from app.model.BasicModel import BasicModel from app.model.BasicModel import BasicModel
from app.model.ProjectModel import ProjectModel from app.model.ProjectModel import ProjectModel
from app.model.ReimburseTravelModel import ReimburseTravelModel
from app.utils.create_module_service import create_model_service from app.utils.create_module_service import create_model_service
@@ -16,6 +17,8 @@ class ReimburseModel(BasicModel, table=True):
remarks: str = Field(default=None, description="备注信息") remarks: str = Field(default=None, description="备注信息")
amount: float = Field(default=None, description="报销金额") amount: float = Field(default=None, description="报销金额")
# /*---------------------------------------project-------------------------------------------*/
proj_id: str = Field( proj_id: str = Field(
default=None, default=None,
description="所属项目id", description="所属项目id",
@@ -40,12 +43,34 @@ class ReimburseModel(BasicModel, table=True):
def project(self, value: Optional[ProjectModel]) -> None: def project(self, value: Optional[ProjectModel]) -> None:
pass pass
# /*---------------------------------------travel_list-------------------------------------------*/
# 新增:与ReimburseTravelModel的一对多关系
travel_list_rel: Optional[List["ReimburseTravelModel"]] = Relationship(
sa_relationship_kwargs={
"foreign_keys": "ReimburseTravelModel.reimburse_id",
"primaryjoin": "ReimburseModel.id == ReimburseTravelModel.reimburse_id",
}
)
@computed_field
@property
def travel_list(self) -> Optional[List["ReimburseTravelModel"]]:
return self.travel_list_rel
@travel_list.setter
def travel_list(self, value: Optional[List["ReimburseTravelModel"]]) -> None:
pass
ReimburseService = create_model_service( ReimburseService = create_model_service(
Cls=ReimburseModel, Cls=ReimburseModel,
custom_query=lambda: select(ReimburseModel) custom_query=lambda: select(ReimburseModel)
.options( .options(
selectinload(ReimburseModel.proj_rel). selectinload(ReimburseModel.proj_rel).
selectinload(ProjectModel.leader) selectinload(ProjectModel.leader),
# 新增:预加载差旅明细
selectinload(ReimburseModel.travel_list_rel)
) )
) )