from dataclasses import dataclass
from typing import Optional, List
from enum import Enum

from external_api.base.search_models import PlaylistImageModel


class PlaylistProvider(Enum):
    Spotify = "spotify"
    Apple = "apple"


@dataclass
class PlaylistsQuery:
    query: Optional[str]
    limit: int


@dataclass
class PlaylistModel:
    id: str
    name: Optional[str]
    owner_name: Optional[str]
    images: List[PlaylistImageModel]


@dataclass
class PlaylistResults:
    playlists: List[PlaylistModel]
    provider: PlaylistProvider