from enum import Enum

from apollo_main_db.push_notifications.models import PushMessage, PushMessageReduced


class PushSearchMode(Enum):
    """Table to check existing push messages in.
    Reduced - save data to PushMessageReduced only.
    Historical - save data to PushMessage only.
    """
    REDUCED = "REDUCED"
    HISTORICAL = "HISTORICAL"


PUSH_SEARCH_MODE_TO_MODEL = {
    PushSearchMode.REDUCED: PushMessageReduced,
    PushSearchMode.HISTORICAL: PushMessage,
}


class DSP:
    SPOTIFY = "spotify"
    APPLE = "apple"


GLOBAL_MARKET: str = 'global'
GLOBAL_MARKET_ID: int = 14
APOLLO_GLOBAL_MARKET: str = '_gl'
US_MARKET: str = 'us'
US_MARKET_ID: int = 4
APPLE_CHART_NAME: str = 'track_chart_playlist'
SPOTIFY_URL_PREFIX = 'https://open.spotify.com/track/'
SPOTIFY_TRACK_PREFIX: str = 'spotify:track:'

APPLE_CATEGORY_ID: int = 11
SONY_CATEGORY_ID: int = 1
SPOTIFY_CATEGORY_ID: int = 5

PLAYLIST_ENTITY_TYPE = "playlist"
SPOTIFY_PLAYLIST_IMAGE_URL_MASK = "https://dhfsk7jl2g0km.cloudfront.net/playlist-{playlist_id}.jpeg"

DATETIME_FORMAT = "%Y-%m-%d %H:%M:%S"


class ReplicaType(Enum):
    """Replicated db type."""
    MASTER = "master"
    SLAVE = "slave"