| """Business logic services.""" |
|
|
| from __future__ import annotations |
|
|
| from .repo import CaseAssignmentRepo |
|
|
|
|
| class CaseAssignmentAggregationService: |
| def __init__(self, case_assignment_repo: CaseAssignmentRepo | None = None) -> None: |
| self._case_assignment_repo = case_assignment_repo |
|
|
| @property |
| def case_assignment_repo(self) -> CaseAssignmentRepo: |
| if self._case_assignment_repo is None: |
| self._case_assignment_repo = CaseAssignmentRepo() |
| return self._case_assignment_repo |
|
|
| def get_primary_lawyer_names_by_case_ids(self, case_ids: list[int]) -> dict[int, str | None]: |
| if not case_ids: |
| return {} |
|
|
| assignments = self.case_assignment_repo.list_assignments_by_case_ids(case_ids) |
|
|
| result: dict[int, str | None] = {} |
| for assignment in assignments: |
| if assignment.case_id in result: |
| continue |
| lawyer = getattr(assignment, "lawyer", None) |
| if lawyer: |
| result[assignment.case_id] = lawyer.real_name or lawyer.username |
| else: |
| result[assignment.case_id] = None |
|
|
| for case_id in case_ids: |
| result.setdefault(case_id, None) |
|
|
| return result |
|
|
