"""Schema for top countries streams endpoint."""

from marshmallow import fields

from sound_recordings.schemas.common import (
    BaseSchema,
    SourceSchema,
    TopCountriesDownloadsBreakdownSchema,
    TopCountriesStreamsBreakdownSchema,
)


class TopCountriesStreamsSchema(BaseSchema):
    """Schema for Top Countries by Streams."""

    isrc = fields.String()
    top_countries = fields.Nested(TopCountriesStreamsBreakdownSchema, many=True)
    sources = fields.Nested(SourceSchema, many=True)


class TopCountriesDownloadsSchema(BaseSchema):
    """Schema for Top Countries by Downloads."""

    isrc = fields.String()
    top_countries = fields.Nested(TopCountriesDownloadsBreakdownSchema, many=True)
    sources = fields.Nested(SourceSchema, many=True)