diff --git a/spotify_actions/search.py b/spotify_actions/search.py index 9d301e0..24c13c6 100644 --- a/spotify_actions/search.py +++ b/spotify_actions/search.py @@ -13,16 +13,24 @@ from .util import exhaust class Query: "Query builder for Spotify search API" - def __init__(self, query: Optional[str] = None, artist: Optional[str] = None, label: Optional[str] = None) -> None: + def __init__( + self, + query: Optional[str] = None, + artist: Optional[str] = None, + label: Optional[str] = None, + year: Optional[int] = None, + ) -> None: self.query = query self.artist = artist self.label = label + self.year = year def __str__(self) -> str: query_items = [ self.query, f'artist:"{self.artist}"' if self.artist is not None else None, f'label:"{self.label}"' if self.label is not None else None, + f"year:{self.year}" if self.year is not None else None, ] return " ".join([i for i in query_items if i is not None])