diff --git a/src/album.py b/src/album.py index 12f86eb..19004d1 100644 --- a/src/album.py +++ b/src/album.py @@ -35,7 +35,13 @@ def get_artist_albums(artist_id): """ Returns artist's albums """ resp = ZSpotify.invoke_url(f'{ARTIST_URL}/{artist_id}/albums') # Return a list each album's id - return [resp[ITEMS][i][ID] for i in range(len(resp[ITEMS]))] + album_ids = [resp[ITEMS][i][ID] for i in range(len(resp[ITEMS]))] + # Recursive requests to get all albums including singles an EPs + while resp['next'] is not None: + resp = ZSpotify.invoke_url(resp['next']) + album_ids.extend([resp[ITEMS][i][ID] for i in range(len(resp[ITEMS]))]) + + return album_ids def download_album(album):