| """Business logic services.""" |
|
|
| from __future__ import annotations |
|
|
| from typing import Any, ClassVar |
|
|
| from django.utils.translation import gettext_lazy as _ |
|
|
| from apps.contracts.models import Contract |
| from apps.core.exceptions import NotFoundError |
| from apps.core.models.enums import CaseType |
|
|
| from ..wiring import get_case_service |
|
|
|
|
| class ContractAdminQueryService: |
| CASE_ALLOWED_TYPES: ClassVar = { |
| CaseType.CIVIL, |
| CaseType.CRIMINAL, |
| CaseType.ADMINISTRATIVE, |
| CaseType.LABOR, |
| CaseType.INTL, |
| } |
|
|
| def can_create_case(self, contract_id: int) -> bool: |
| contract = Contract.objects.filter(pk=contract_id).only("id", "case_type").first() |
| if not contract: |
| return False |
| return contract.case_type in self.CASE_ALLOWED_TYPES |
|
|
| def get_contract_detail(self, contract_id: int) -> Any: |
| try: |
| return ( |
| Contract.objects.select_related("folder_binding") |
| .prefetch_related( |
| "contract_parties__client", |
| "assignments__lawyer", |
| "payments", |
| "reminders", |
| "supplementary_agreements__parties__client", |
| "finalized_materials", |
| ) |
| .get(pk=contract_id) |
| ) |
| except Contract.DoesNotExist as exc: |
| raise NotFoundError(_("合同 %(id)s 不存在") % {"id": contract_id}) from exc |
|
|
| def get_related_cases(self, contract_id: int) -> list[dict[str, Any]]: |
| case_service = get_case_service() |
| cases_dto = case_service.get_cases_by_contract(contract_id) |
| if not cases_dto: |
| return [] |
|
|
| case_ids = [case.id for case in cases_dto] |
| case_primary_lawyer_map = case_service.get_primary_lawyer_names_by_case_ids_internal(case_ids) |
|
|
| from apps.core.models.enums import CaseStatus |
|
|
| status_map = dict(CaseStatus.choices) |
| return [ |
| { |
| "id": case.id, |
| "name": case.name, |
| "status": case.status, |
| "status_display": status_map.get(case.status, case.status), |
| "cause_of_action": case.cause_of_action or _("未设置"), |
| "primary_lawyer": case_primary_lawyer_map.get(case.id) or _("未指派"), |
| "detail_url": f"/admin/cases/case/{case.id}/detail/", |
| } |
| for case in cases_dto |
| ] |
|
|
