"""Statement Period Payment Entity Marshmallow schema."""

from abacus_common_logic.marshalling.custom_fields import ma


class StatementPeriodPaymentEntitySchema(ma.Schema):
    """Statement Period Payment Entity schema."""

    statement_period_payment_entity_id = ma.IntegerId()
    statement_period_id = ma.IntegerId(required=True)
    reference_payment_entity_id = ma.IntegerId(required=True)
    is_visible_to_customer = ma.Boolean(required=True)


class StatementPeriodPaymentEntityAbacusStateSchema(ma.Schema):
    """Statement Period Payment Entity abacus_state schema."""

    statement_period_payment_entity_id = ma.IntegerId()
    reference_payment_entity_id = ma.IntegerId(required=True)
    statement_period_id = ma.IntegerId(required=True)
    abacus_state_id = ma.NonNegativeInteger(required=True)
    action_name = ma.NonemptyString(required=True)
    action_status = ma.NonemptyString(required=True)