from enum import Enum

from delphi_api.v2.constants import COLUMN_FAMILY_ID


class DateConfig(Enum):
    """Misc. date configuration constants"""
    FormatString = '%Y-%m-%d'


class TableName(Enum):
    """Database table names"""
    # Amazon
    amazon_stream_playlist = 'amazon_stream_playlist'
    amazon_stream_playlist_track = 'amazon_stream_playlist_track'
    amazon_stream_track = 'amazon_stream_track'

    # Apple
    apple_stream_track = 'apple_stream_track'

    # Spotify
    spotify_stream_track = 'spotify_stream_track'

    # YouTube
    youtube_stream_track = 'youtube_stream_track'


class ColumnFamily(Enum):
    AMAZON = COLUMN_FAMILY_ID
    APPLE = COLUMN_FAMILY_ID
    SPOTIFY = COLUMN_FAMILY_ID
    YOUTUBE = COLUMN_FAMILY_ID


class ColumnName(Enum):
    DATE = 'report_date'
    DSP = 'dsp'
    ISRC = 'isrc'
    PLAYLIST_ID = 'playlist_id'