| """Case admin export bridge helpers.""" |
|
|
| from __future__ import annotations |
|
|
| from typing import TYPE_CHECKING, Callable |
|
|
| if TYPE_CHECKING: |
| from apps.cases.models import Case |
|
|
| _CASE_ADMIN_EXPORT_PREFETCHES: tuple[str, ...] = ( |
| "parties__client__identity_docs", |
| "parties__client__property_clues__attachments", |
| "assignments__lawyer", |
| "supervising_authorities", |
| "case_numbers", |
| "chats", |
| "logs__actor", |
| "logs__attachments", |
| ) |
|
|
| _CASE_ADMIN_FILE_PREFETCHES: tuple[str, ...] = ( |
| "parties__client__identity_docs", |
| "parties__client__property_clues__attachments", |
| "logs__attachments", |
| ) |
|
|
|
|
| def get_case_admin_export_prefetches() -> tuple[str, ...]: |
| """Return case-side prefetch paths needed by CaseAdmin export.""" |
| return _CASE_ADMIN_EXPORT_PREFETCHES |
|
|
|
|
| def get_case_admin_file_prefetches() -> tuple[str, ...]: |
| """Return case-side prefetch paths needed by CaseAdmin file export.""" |
| return _CASE_ADMIN_FILE_PREFETCHES |
|
|
|
|
| def collect_case_file_paths_for_export( |
| case: Case, |
| add_path: Callable[[str], None], |
| ) -> None: |
| """Collect case-side file paths for admin export.""" |
| for party in case.parties.all(): |
| for identity_doc in party.client.identity_docs.all(): |
| add_path(identity_doc.file_path) |
| for clue in party.client.property_clues.all(): |
| for attachment in clue.attachments.all(): |
| add_path(attachment.file_path) |
|
|
| for log in case.logs.all(): |
| for attachment in log.attachments.all(): |
| if attachment.file: |
| add_path(attachment.file.name) |
|
|
