"""Legacy contract marshmallow schema."""

from abacus_common_logic.marshalling.custom_fields import ma


class LegacyContractSchema(ma.Schema):
    """Legacy contract schema."""

    legacy_contract_id = ma.IntegerId()
    contract_id = ma.NonNegativeInteger(required=True)
    oa_contract_id = ma.NonNegativeInteger(allow_none=True)