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

from marshmallow import fields

from analytics.schemas.common import BaseSchema, SourceSchema


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

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


class TopCountriesVideosSchema(BaseSchema):
    """Schema for Top Countries by video views."""

    video_id = fields.String()
    top_countries_videos = fields.Nested(TopCountriesVideosBreakdownSchema, many=True)
    sources = fields.Nested(SourceSchema, many=True)