From 34584eadd1e962576aeff33b23b2ba43b528b7a1 Mon Sep 17 00:00:00 2001 From: Andrii Kohut Date: Mon, 25 Aug 2014 19:14:41 +0300 Subject: [PATCH] Works faster then checking each result item with isinstance --- beetsplug/discogs.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/beetsplug/discogs.py b/beetsplug/discogs.py index 0301cb67e..75fe536c4 100644 --- a/beetsplug/discogs.py +++ b/beetsplug/discogs.py @@ -98,13 +98,8 @@ class DiscogsPlugin(BeetsPlugin): # Strip medium information from query, Things like "CD1" and "disk 1" # can also negate an otherwise positive result. query = re.sub(r'(?i)\b(CD|disc)\s*\d+', '', query) - albums = [] - for result in self.discogs_client.search(query): - if isinstance(result, Release): - albums.append(self.get_album_info(result)) - if len(albums) >= 5: - break - return albums + releases = self.discogs_client.search(query, type='release').page(1) + return [self.get_album_info(release) for release in releases[:5]] def get_album_info(self, result): """Returns an AlbumInfo object for a discogs Release object.