| """Business logic services.""" | |
| from __future__ import annotations | |
| from typing import Any | |
| from apps.core.dto import CaseDTO | |
| class CaseDtoAssembler: | |
| def to_dto(self, case: Any, case_number: str | None = None) -> CaseDTO: | |
| dto = CaseDTO.from_model(case) | |
| dto.case_number = case_number | |
| return dto | |
| def to_dtos(self, cases: list[Any], case_number_map: dict[int, str | None] | None = None) -> list[CaseDTO]: | |
| case_number_map = case_number_map or {} | |
| return [self.to_dto(case, case_number_map.get(case.id)) for case in cases] | |