"""Schema for payment_group_payment_batch serialization."""

from abacus_common_logic.marshalling.custom_fields import ma


class PaymentGroupPaymentBatchSchema(ma.Schema):
    """Payment Group Payment Batch schema."""

    payment_group_payment_batch_id = ma.NonNegativeInteger(required=True)
    payment_group_payment_id = ma.NonNegativeInteger(required=True)
    payoneer_program_id = ma.NonNegativeInteger(required=True)
    batch_num = ma.NonNegativeInteger(required=True)


class PaymentGroupPaymentBatchStatusSchema(PaymentGroupPaymentBatchSchema):
    """Payment Group Payment Batch Status schema."""

    abacus_state_id = ma.NonNegativeInteger(required=True)
    action_status = ma.NonemptyString(required=True)
    error_message = ma.NonemptyString()