| """当事人 DTO 组装器。""" |
|
|
| from __future__ import annotations |
|
|
| from typing import TYPE_CHECKING |
|
|
| from apps.core.dto import ClientIdentityDocDTO, PropertyClueDTO |
| from apps.core.interfaces import ClientDTO |
|
|
| if TYPE_CHECKING: |
| from apps.client.models import Client, ClientIdentityDoc, PropertyClue |
|
|
|
|
| class ClientDtoAssembler: |
| """当事人基础 DTO 组装。""" |
|
|
| def to_dto(self, client: Client) -> ClientDTO: |
| return ClientDTO( |
| id=client.id, |
| name=client.name, |
| client_type=client.client_type, |
| phone=client.phone, |
| id_number=client.id_number, |
| address=client.address, |
| is_our_client=client.is_our_client, |
| ) |
|
|
|
|
| class ClientRelatedDtoAssembler: |
| """当事人关联对象 DTO 组装。""" |
|
|
| def property_clue_to_dto(self, clue: PropertyClue) -> PropertyClueDTO: |
| return PropertyClueDTO( |
| id=clue.pk, |
| client_id=clue.client_id, |
| clue_type=clue.clue_type, |
| content=clue.content, |
| description=None, |
| ) |
|
|
| def property_clues_to_dtos(self, clues: list[PropertyClue]) -> list[PropertyClueDTO]: |
| return [self.property_clue_to_dto(c) for c in clues] |
|
|
| def identity_doc_to_dto(self, doc: ClientIdentityDoc) -> ClientIdentityDocDTO: |
| return ClientIdentityDocDTO( |
| id=doc.pk, |
| client_id=doc.client_id, |
| doc_type=doc.doc_type, |
| doc_type_display=doc.get_doc_type_display(), |
| file_path=doc.media_url, |
| expiry_date=str(doc.expiry_date) if doc.expiry_date else None, |
| is_valid=True, |
| ) |
|
|
| def identity_docs_to_dtos(self, docs: list[ClientIdentityDoc]) -> list[ClientIdentityDocDTO]: |
| return [self.identity_doc_to_dto(d) for d in docs] |
|
|
