finalize medical consultation agent backend
This commit is contained in:
@@ -3,7 +3,7 @@ from sqlalchemy.orm import Session, selectinload
|
||||
|
||||
from app.models.source_case import CaseBase, CaseExamItem, ScoringRule, TeachingCase, TraditionalCase
|
||||
from app.models.training import SessionOrder, SessionSubmission, TrainingSession
|
||||
from app.models.training_record import TrainingRecord
|
||||
from app.models.training_record import TrainingRecord, TrainingScoreDetail
|
||||
|
||||
|
||||
class CaseRepository:
|
||||
@@ -64,6 +64,7 @@ class CaseRepository:
|
||||
"training_session": len(session_ids),
|
||||
"training_order": self._count_training_orders(case_id, session_ids),
|
||||
"training_submission": self._count_by_sessions(SessionSubmission, SessionSubmission.session_id, session_ids),
|
||||
"training_score_detail": self._count_score_details(case_id, session_ids),
|
||||
"training_record": self._count_training_records(case_id, session_ids),
|
||||
}
|
||||
|
||||
@@ -75,6 +76,7 @@ class CaseRepository:
|
||||
deleted["training_submission"] = self._delete_by_sessions(
|
||||
SessionSubmission, SessionSubmission.session_id, session_ids
|
||||
)
|
||||
deleted["training_score_detail"] = self._delete_score_details(case_id, session_ids)
|
||||
deleted["training_record"] = self._delete_training_records(case_id, session_ids)
|
||||
deleted["training_session"] = self._delete_where(TrainingSession, TrainingSession.case_id == case_id)
|
||||
deleted["case_exam_item"] = self._delete_where(CaseExamItem, CaseExamItem.case_id == case_id)
|
||||
@@ -115,6 +117,13 @@ class CaseRepository:
|
||||
)
|
||||
return self._count(TrainingRecord, TrainingRecord.case_id == case_id)
|
||||
|
||||
def _count_score_details(self, case_id: int, session_ids: list[int]) -> int:
|
||||
"""病例删除预览:统计该病例评价记录下的评分明细。"""
|
||||
record_ids = self._record_ids(case_id, session_ids)
|
||||
if not record_ids:
|
||||
return 0
|
||||
return self._count(TrainingScoreDetail, TrainingScoreDetail.record_id.in_(record_ids))
|
||||
|
||||
def _delete_where(self, model: type, *criteria) -> int:
|
||||
"""病例删除执行:按条件删除单表记录并返回影响行数。"""
|
||||
result = self.db.execute(delete(model).where(*criteria))
|
||||
@@ -144,6 +153,23 @@ class CaseRepository:
|
||||
)
|
||||
return self._delete_where(TrainingRecord, TrainingRecord.case_id == case_id)
|
||||
|
||||
def _delete_score_details(self, case_id: int, session_ids: list[int]) -> int:
|
||||
"""病例删除执行:先删除评价明细,避免阻塞训练记录删除。"""
|
||||
record_ids = self._record_ids(case_id, session_ids)
|
||||
if not record_ids:
|
||||
return 0
|
||||
return self._delete_where(TrainingScoreDetail, TrainingScoreDetail.record_id.in_(record_ids))
|
||||
|
||||
def _record_ids(self, case_id: int, session_ids: list[int]) -> list[int]:
|
||||
"""病例删除:读取该病例关联的训练记录 ID 集合。"""
|
||||
if session_ids:
|
||||
stmt = select(TrainingRecord.id).where(
|
||||
or_(TrainingRecord.case_id == case_id, TrainingRecord.session_id.in_(session_ids))
|
||||
)
|
||||
else:
|
||||
stmt = select(TrainingRecord.id).where(TrainingRecord.case_id == case_id)
|
||||
return [int(item) for item in self.db.scalars(stmt).all()]
|
||||
|
||||
def get_exam_items(self, case_id: int) -> list[CaseExamItem]:
|
||||
"""检查项目:读取当前病例下全部可申请检查检验项目。"""
|
||||
stmt = select(CaseExamItem).where(CaseExamItem.case_id == case_id).order_by(CaseExamItem.display_order)
|
||||
|
||||
Reference in New Issue
Block a user