"""Pydantic models for Lambda response validation."""

from pydantic import BaseModel, Field


class PaymentAllocationResponse(BaseModel):
    """Response model for payment allocation processing."""

    statement_period_id: int = Field(..., description='ID of the statement period')
    statement_period_payment_entity_id: int = Field(
        ..., description='ID of the statement period payment entity'
    )
    allocations_created: int = Field(
        ..., description='Number of payment_allocation records created'
    )
    ledger_adjustments_linked: int = Field(
        ...,
        description='Number of payment_allocation_ledger_adjustment records created',
    )