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)
|
||||
|
||||
@@ -1,7 +1,7 @@
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy import delete, select
|
||||
from sqlalchemy.orm import Session
|
||||
|
||||
from app.models.training_record import TrainingRecord
|
||||
from app.models.training_record import TrainingRecord, TrainingScoreDetail
|
||||
|
||||
|
||||
class EvaluationRepository:
|
||||
@@ -16,6 +16,20 @@ class EvaluationRepository:
|
||||
self.db.flush()
|
||||
return record
|
||||
|
||||
def replace_score_details(self, record_id: int, details: list[TrainingScoreDetail]) -> list[TrainingScoreDetail]:
|
||||
"""评分明细保存:按训练记录覆盖写入维度评分明细。"""
|
||||
self.db.execute(delete(TrainingScoreDetail).where(TrainingScoreDetail.record_id == record_id))
|
||||
for detail in details:
|
||||
detail.record_id = record_id
|
||||
self.db.add(detail)
|
||||
self.db.flush()
|
||||
return details
|
||||
|
||||
def list_score_details(self, record_id: int) -> list[TrainingScoreDetail]:
|
||||
"""评分明细读取:按训练记录查询全部维度明细。"""
|
||||
stmt = select(TrainingScoreDetail).where(TrainingScoreDetail.record_id == record_id).order_by(TrainingScoreDetail.id)
|
||||
return list(self.db.scalars(stmt).all())
|
||||
|
||||
def get_by_session(self, session_id: int, user_id: str) -> TrainingRecord | None:
|
||||
"""评价读取:按会话 ID 和外部 user_id 查询训练记录。"""
|
||||
stmt = select(TrainingRecord).where(
|
||||
|
||||
@@ -1,25 +0,0 @@
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.orm import Session
|
||||
|
||||
from app.models.user import UserLearningProfile
|
||||
|
||||
|
||||
class UserLearningProfileRepository:
|
||||
"""学习档案仓储:维护用户训练评价聚合数据。"""
|
||||
|
||||
def __init__(self, db: Session) -> None:
|
||||
self.db = db
|
||||
|
||||
def get_profile(self, user_id: str, tenant_id: str | None) -> UserLearningProfile | None:
|
||||
"""档案读取:按 user_id 和 tenant_id 获取学习档案。"""
|
||||
stmt = select(UserLearningProfile).where(
|
||||
UserLearningProfile.user_id == user_id,
|
||||
UserLearningProfile.tenant_id == tenant_id,
|
||||
)
|
||||
return self.db.scalar(stmt)
|
||||
|
||||
def save(self, profile: UserLearningProfile) -> UserLearningProfile:
|
||||
"""档案保存:创建或更新用户学习档案。"""
|
||||
self.db.add(profile)
|
||||
self.db.flush()
|
||||
return profile
|
||||
@@ -50,7 +50,7 @@ class SourceCaseRepository:
|
||||
return self.db.scalar(stmt)
|
||||
|
||||
def get_department_name(self, department_id: int | None) -> str:
|
||||
"""科室名称:兼容当前 demo 的 departments 表,源库无科室表时返回空字符串。"""
|
||||
"""科室名称:按用户端 department 表读取科室名称。"""
|
||||
if not department_id:
|
||||
return ""
|
||||
department = self.db.scalar(select(Department).where(Department.id == department_id))
|
||||
|
||||
Reference in New Issue
Block a user