| """Module for client.""" |
|
|
| from __future__ import annotations |
|
|
| from dataclasses import dataclass |
| from typing import TYPE_CHECKING |
|
|
| if TYPE_CHECKING: |
| from apps.client.models import Client, ClientIdentityDoc |
|
|
|
|
| @dataclass |
| class ClientDTO: |
| id: int |
| name: str |
| client_type: str |
| phone: str | None = None |
| id_number: str | None = None |
| address: str | None = None |
| is_our_client: bool = False |
|
|
| @classmethod |
| def from_model(cls, client: Client) -> ClientDTO: |
| return cls( |
| 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, |
| ) |
|
|
|
|
| @dataclass |
| class PropertyClueDTO: |
| id: int |
| client_id: int |
| clue_type: str |
| content: str |
| description: str | None = None |
|
|
|
|
| @dataclass |
| class ClientIdentityDocDTO: |
| id: int |
| client_id: int |
| doc_type: str |
| doc_type_display: str |
| file_path: str | None = None |
| expiry_date: str | None = None |
| is_valid: bool = True |
|
|
| @classmethod |
| def from_model(cls, doc: ClientIdentityDoc) -> ClientIdentityDocDTO: |
| return cls( |
| id=doc.id, |
| client_id=doc.client_id, |
| doc_type=doc.doc_type, |
| doc_type_display=doc.get_doc_type_display(), |
| file_path=doc.media_url, |
| ) |
|
|
