"""Marshmallow schema for the country_attribution pipeline.

Used via `--schema pipelines.country_attribution.schemas:CountryAttributionSchema`
to coerce CSV string values into the native types ows-account's
`UpdateVendorCountryIdSchema` expects (country_id: Integer).
"""

from marshmallow import Schema, fields


class CountryAttributionSchema(Schema):  # type: ignore[misc]
    vendor_uuid = fields.String(required=True)
    country_id = fields.Integer(required=True)