| """Business logic services.""" | |
| from __future__ import annotations | |
| from apps.contracts.models import Contract, ContractAssignment | |
| class ContractAssignmentQueryService: | |
| def list_lawyer_ids_by_contract_internal(self, contract_id: int) -> list[int]: | |
| if not Contract.objects.filter(id=contract_id).exists(): | |
| return [] | |
| return list( | |
| ContractAssignment.objects.filter(contract_id=contract_id) | |
| .order_by("-is_primary", "order", "id") | |
| .values_list("lawyer_id", flat=True) | |
| ) | |
| def get_primary_lawyer(self, contract_id: int) -> ContractAssignment | None: | |
| """获取合同的主办律师指派记录。""" | |
| return ( | |
| ContractAssignment.objects.filter(contract_id=contract_id, is_primary=True).select_related("lawyer").first() | |
| ) | |
| def get_all_lawyers(self, contract_id: int) -> list[ContractAssignment]: | |
| """获取合同的所有律师指派记录（按 is_primary 降序、order 升序）。""" | |
| return list( | |
| ContractAssignment.objects.filter(contract_id=contract_id) | |
| .select_related("lawyer") | |
| .order_by("-is_primary", "order", "id") | |
| ) | |