"""Query class for sound recording top markets."""

from marshmallow import fields

from analytics.connectors.snowflake import AbstractSnowflakeQuery
from analytics.queries.schema import QueryWithPermissionsSchema


class SoundRecordingTopMarketsQuerySchema(QueryWithPermissionsSchema):
    isrc = fields.String(required=True)
    store_ids = fields.List(fields.Int, required=True)
    distributors = fields.List(fields.String, required=True)
    country_ids = fields.List(fields.String, load_default=[])


class TopMarkets(AbstractSnowflakeQuery):
    query_schema = SoundRecordingTopMarketsQuerySchema
    filename = "sound_recording/top_markets.sql"