from enum import Enum
from typing import List


APP_NAME = "download_spotify_charts"

ENV_PROD = "prod"


class ExtEnum(Enum):
    @classmethod
    def values(cls) -> List[str]:
        return [i.value for i in cls]

    @classmethod
    def all_str(cls):
        return ",".join(i.value for i in cls)


WORLDWIDE = "worldwide"
GLOBAL = "global"


class ChartBreakdown(ExtEnum):
    WEEKLY = "weekly"
    DAILY = "daily"


class ChartType(ExtEnum):
    REGIONAL = "regional"
    VIRAL = "viral"


CSV_NOTE = (
    "",
    "",
    "",
    (
        "Note that these figures are generated using a formula that protects against any artificial inflation of " +
        "chart positions."
    ),
    "",
)

CSV_HEADER = {
    ChartType.REGIONAL: ("Position", "Track Name", "Artist", "Streams", "URL"),
    ChartType.VIRAL: ("Position", "Track Name", "Artist", "URL"),
}