from typing import List

from apollo_utils.job.clients.base.client import BaseApiClient

from juno_email_messages import config
from juno_email_messages.clients.base_config import ApiConfig
from juno_email_messages.constants import Service


class GateClientConfig(ApiConfig):
    service = Service.GATE


class GateClient(GateClientConfig, BaseApiClient):
    def get_juno_tracks(self, **params) -> List[dict]:
        return self.send_request(
            "api/charts/tracks/streams/spotify/",
            get_json=True,
            limit=config.JUNO_TRACK_COUNT,
            sort_by="streams_days_7",
            sort_order="desc",
            distributors="awal,theorchard,sme",
            **params,
        )