"""Abacus statement period schemas."""

from collaborator.schemas import BaseSchema
from collaborator.schemas.dp_payment import DpPaymentAction, DpPaymentSchema


class GetDpPaymentsForAbacusStatementPeriodResponseSchema(BaseSchema):
    """GET dp_payments for Abacus statement period response schema."""

    currency_agnostic_total_amount: float
    total_count: int
    payments: list[DpPaymentSchema]


class PutDpPaymentsForAbacusStatementPeriodResponseSchema(BaseSchema):
    """PUT dp_payments for Abacus statement period response schema."""

    currency_agnostic_total_amount: float
    total_count: int
    payments: list[DpPaymentSchema]


class PatchDpPaymentsForAbacusStatementPeriodSchema(BaseSchema):
    """PATCH dp_payments for Abacus statement period schema."""

    action_name: DpPaymentAction


class PatchDpPaymentsForAbacusStatementPeriodResponseSchema(BaseSchema):
    """PATCH dp_payments for Abacus statement period response schema."""

    total_count: int
    payments: list[DpPaymentSchema]