feat: 关联查询出来报销单的差旅费用子数据
This commit is contained in:
@@ -1,4 +1,4 @@
|
||||
from typing import Optional
|
||||
from typing import Optional, List
|
||||
|
||||
from pydantic import computed_field
|
||||
from sqlalchemy.orm import selectinload
|
||||
@@ -6,6 +6,7 @@ from sqlmodel import Field, Relationship, select
|
||||
|
||||
from app.model.BasicModel import BasicModel
|
||||
from app.model.ProjectModel import ProjectModel
|
||||
from app.model.ReimburseTravelModel import ReimburseTravelModel
|
||||
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="备注信息")
|
||||
amount: float = Field(default=None, description="报销金额")
|
||||
|
||||
# /*---------------------------------------project-------------------------------------------*/
|
||||
|
||||
proj_id: str = Field(
|
||||
default=None,
|
||||
description="所属项目id",
|
||||
@@ -40,12 +43,34 @@ class ReimburseModel(BasicModel, table=True):
|
||||
def project(self, value: Optional[ProjectModel]) -> None:
|
||||
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(
|
||||
Cls=ReimburseModel,
|
||||
custom_query=lambda: select(ReimburseModel)
|
||||
.options(
|
||||
selectinload(ReimburseModel.proj_rel).
|
||||
selectinload(ProjectModel.leader)
|
||||
selectinload(ProjectModel.leader),
|
||||
|
||||
# 新增:预加载差旅明细
|
||||
selectinload(ReimburseModel.travel_list_rel)
|
||||
)
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user