From e9477a31013c3b888958e309399b22446d86276b Mon Sep 17 00:00:00 2001 From: Kepoor Hampond Date: Mon, 16 Jan 2017 13:18:58 -0800 Subject: [PATCH] Added a no results message --- irs/manager.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/irs/manager.py b/irs/manager.py index bbcfbf1..1eb2bbf 100644 --- a/irs/manager.py +++ b/irs/manager.py @@ -190,6 +190,7 @@ class Manager: def get_album_contents(self, search): spotify = spotipy.Spotify() + results = spotify.search(q=search, type='album') items = results['albums']['items'] if len(items) > 0: @@ -210,6 +211,9 @@ class Manager: song = song.split(" - ")[0] names.append(song) return names, album_id + else: + print (bc.FAIL + "No results were found for your album search." + bc.ENDC) + exit(1) def get_album_art(self, artist, album, id=None): spotify = spotipy.Spotify() @@ -227,7 +231,9 @@ class Manager: def rip_album(self): search = self.args.artist + " " + self.args.album - songs, album_uri = self.get_album_contents(search) + print(self.get_album_contents(search)) + import time + time.sleep(10) if not songs: print (bc.FAIL + "Could not find album." + bc.ENDC)