"""Schema for top countries views endpoint."""

from marshmallow import fields

from analytics.schemas.common import BaseSchema, SourceSchema


class TopCountriesChannelsBreakdownSchema(BaseSchema):
    """Schema for top countries with downloads."""

    country_code = fields.String()
    views = fields.Integer()


class TopCountriesChannelsSchema(BaseSchema):
    """Schema for Top Countries by channel views."""

    channel_id = fields.String()
    top_countries_channels = fields.Nested(
        TopCountriesChannelsBreakdownSchema, many=True
    )
    sources = fields.Nested(SourceSchema, many=True)