| """Module for cases.""" |
|
|
| from __future__ import annotations |
|
|
| from dataclasses import dataclass, field |
| from typing import TYPE_CHECKING |
|
|
| if TYPE_CHECKING: |
| from apps.cases.models import Case, CaseParty, CaseTemplateBinding |
|
|
|
|
| @dataclass |
| class CaseDTO: |
| id: int |
| name: str |
| current_stage: str | None = None |
| contract_id: int | None = None |
| status: str = "active" |
| case_type: str | None = None |
| cause_of_action: str | None = None |
| target_amount: str | None = None |
| is_archived: bool = False |
| start_date: str | None = None |
| effective_date: str | None = None |
| case_number: str | None = None |
|
|
| @classmethod |
| def from_model(cls, case: Case) -> CaseDTO: |
| return cls( |
| id=case.id, |
| name=case.name, |
| current_stage=case.current_stage, |
| contract_id=case.contract_id, |
| status=case.status, |
| case_type=case.case_type, |
| cause_of_action=case.cause_of_action, |
| target_amount=str(case.target_amount) if case.target_amount is not None else None, |
| is_archived=case.is_archived, |
| start_date=str(case.start_date) if case.start_date else None, |
| effective_date=str(case.effective_date) if case.effective_date else None, |
| ) |
|
|
|
|
| @dataclass |
| class CaseSearchResultDTO: |
| id: int |
| name: str |
| case_numbers: list[str] = field(default_factory=list) |
| parties: list[str] = field(default_factory=list) |
| created_at: str | None = None |
|
|
|
|
| @dataclass |
| class CaseTemplateBindingDTO: |
| id: int |
| case_id: int |
| template_id: int |
| template_name: str |
| template_function_code: str | None = None |
| binding_source: str = "manual" |
| created_at: str | None = None |
|
|
| @classmethod |
| def from_model(cls, binding: CaseTemplateBinding) -> CaseTemplateBindingDTO: |
| return cls( |
| id=binding.id, |
| case_id=binding.case_id, |
| template_id=binding.template_id, |
| template_name=binding.template.name if binding.template else "", |
| template_function_code=None, |
| binding_source=binding.binding_source, |
| created_at=str(binding.created_at) if binding.created_at else None, |
| ) |
|
|
|
|
| @dataclass |
| class CasePartyDTO: |
| id: int |
| case_id: int |
| client_id: int |
| client_name: str |
| client_type: str |
| legal_status: str |
| id_number: str | None = None |
| address: str | None = None |
| phone: str | None = None |
| legal_representative: str | None = None |
| is_our_client: bool = False |
|
|
| @classmethod |
| def from_model(cls, party: CaseParty) -> CasePartyDTO: |
| client = party.client |
| return cls( |
| id=party.id, |
| case_id=party.case_id, |
| client_id=party.client_id, |
| client_name=client.name if client else "", |
| client_type=client.client_type if client else "natural", |
| legal_status=party.legal_status or "", |
| id_number=client.id_number if client else None, |
| address=client.address if client else None, |
| phone=client.phone if client else None, |
| legal_representative=client.legal_representative if client else None, |
| is_our_client=client.is_our_client if client else False, |
| ) |
|
|
