bspeice/spotify_actions
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add a function for querying labels

Browse Source
This commit is contained in:
Bradlee Speice 2021-07-03 23:04:21 -04:00
parent a783edd3ac
commit 238c052668
2 changed files with 17 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/find_albums.py
View File

@ -5,7 +5,7 @@ Find an album using the `spotify_actions` API and echo its contents.
from argparse import ArgumentParser
from spotify_actions.echo import echo_album
from spotify_actions.search import search_album
from spotify_actions.search import search_albums
from spotify_actions.util import read_credentials
@ -18,7 +18,7 @@ def main() -> None: # pylint: disable=missing-function-docstring
client = read_credentials(cmdline.config)
albums = search_album(client, cmdline.search_string)
albums = search_albums(client, cmdline.search_string)
echo_album(albums)

19
spotify_actions/search.py
View File

@ -9,11 +9,22 @@ from spotipy import Spotify
from .util import exhaust
def search_album(client: Spotify, search_str: str) -> Iterable[SearchAlbum]:
"Display albums from a search string"
def _search_albums(client: Spotify, search_str: str) -> Paging:
def _search(limit: int, offset: int) -> Paging:
return Paging(**client.search(search_str, limit=limit, offset=offset, type="album")["albums"])
for item in exhaust(_search):
return _search
def search_albums(client: Spotify, search_str: str) -> Iterable[SearchAlbum]:
"Display albums from a search string"
for item in exhaust(_search_albums(client, search_str)):
yield SearchAlbum(**item)
def search_label_albums(client: Spotify, search_str: str) -> Iterable[SearchAlbum]:
"Display albums from a particular label"
for item in exhaust(_search_albums(client, f'label:"{search_str}"')):
yield SearchAlbum(**item)