from typing import List, Optional

from pydantic import BaseModel, Field


class KycNotificationCreateResponseModel(BaseModel):
    """KYC notification create model."""

    account_payee_kyc_notification_id: int
    account_payee_id: int
    file_upload_link: str
    requirement_id: Optional[str] = None
    requirement_type_id: Optional[int] = None
    sub_requirement_id: Optional[str] = None
    possible_sub_requirement_types: Optional[str] = None
    sub_requirement_status_id: Optional[int] = None
    entity_reference_id: Optional[str] = None
    entity_reference_type_id: Optional[int] = None


class PayeeItem(BaseModel):
    """Payee item model."""

    payee_id: int


class PayeeItemData(BaseModel):
    """Payee item wrapped in data."""

    data: PayeeItem


class PayeeDataloaderResponseModel(BaseModel):
    """Payee dataloader model."""

    payees: List[PayeeItemData] = Field(min_length=1, max_length=1)