File size: 1,812 Bytes
b4f2238 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 | """当事人 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]
|