from dataclasses import dataclass
from typing import Optional, List
from marshmallow import EXCLUDE


@dataclass
class SearchMediaItem:
    type: str
    apollo_id: str
    title: str
    artist: str
    cover_url: Optional[str]
    external_id: str

    class Meta:
        unknown = EXCLUDE


@dataclass
class SearchResults:
    tracks: List[SearchMediaItem]
    albums: List[SearchMediaItem]