"""Contract Exclusion schemas."""

from abacus_common_logic.marshalling.custom_fields import ma


class ExclusionsSchema(ma.Schema):
    """Schema for contract's exclusions."""

    countries = ma.List(ma.String(), required=True)
    stores = ma.List(ma.String(), required=True)


class ContractExclusionSchema(ma.Schema):
    """Schema for contract exclusion details."""

    contract_exclusion_id = ma.IntegerId(required=True)
    contract_id = ma.IntegerId(required=True)
    exclusions = ma.Nested(ExclusionsSchema)