"""Models for OxaPay Balance API."""

from pydantic import BaseModel, Field

from .common import DataResponse


class AccountBalanceRequest(BaseModel):
    """Request model for account balance."""

    currency: str | None = Field(
        None,
        description="Specific currency to get balance for",
    )


class BalanceInfo(BaseModel):
    """Model for balance information."""

    balance: str = Field(..., description="Balance amount")
    symbol: str = Field(..., description="Currency symbol")


class AccountBalanceResponse(DataResponse[dict[str, BalanceInfo]]):
    """Response model for account balance."""
