"""Pydantic models for database records."""

from decimal import Decimal

from pydantic import BaseModel


class StatementPeriodPaymentEntity(BaseModel):
    """Model for statement_period_payment_entity query result row."""

    statement_period_payment_entity_id: int
    statement_period_id: int
    reference_payment_entity_id: int


class ContractAdjustmentCount(BaseModel):
    """Model for contract-level adjustment count query result."""

    contract_id: int
    adjustment_count: int


class LedgerAdjustmentApplied(BaseModel):
    """Model for ledger_adjustment_applied query result row."""

    ledger_adjustment_applied_id: int
    account_id: int
    contract_id: int
    adjustment_amount: Decimal
    adjustment_currency_code: str
    adjustment_amount_payee_currency: Decimal
    adjustment_payee_currency_code: str
    account_payee_id: int