X-Git-Url: https://git.kaliko.me/?a=blobdiff_plain;f=sima%2Flib%2Fwebserv.py;h=79a9b9b01c58abb342c2eb1e4668809b2cf858f6;hb=40a57f2ddcd2c52318e405383005fe291bd5d9b6;hp=ffd516aae82388e237015daf23114df0279ce662;hpb=1b5cfb4c9187df891bd840b51c044d72ad0077d5;p=mpd-sima.git diff --git a/sima/lib/webserv.py b/sima/lib/webserv.py index ffd516a..79a9b9b 100644 --- a/sima/lib/webserv.py +++ b/sima/lib/webserv.py @@ -33,7 +33,7 @@ from hashlib import md5 from .plugin import Plugin from .track import Track from .meta import Artist -from ..utils.utils import WSError +from ..utils.utils import WSError, WSNotFound def cache(func): """Caching decorator""" @@ -147,7 +147,6 @@ class WebService(Plugin): hist.insert(0, art) reorg = [art for art in alist if art not in hist] reorg.extend(hist) - self.log.info('{}'.format(' / '.join([a.name for a in reorg]))) return reorg @cache @@ -179,6 +178,10 @@ class WebService(Plugin): self.log.debug('Requesting {} for {!r}'.format(self.ws.name, artist)) try: [as_art.append(art) for art in as_artists] + except WSNotFound as err: + if artist.mbid: + return self.ws_similar_artists(Artist(name=artist.name)) + self.log.warning('{}: {}'.format(self.ws.name, err)) except WSError as err: self.log.warning('{}: {}'.format(self.ws.name, err)) if as_art: @@ -193,6 +196,7 @@ class WebService(Plugin): return last_trk = self.player.playlist[-1] extra_arts = list() + ret_extra = list() while depth < self.plugin_conf.getint('depth'): if len(history) == 0: break @@ -203,14 +207,14 @@ class WebService(Plugin): extra_arts.append(trk.Artist) depth += 1 self.log.info('EXTRA ARTS: {}'.format( - '/'.join([art.name for art in extra_arts]))) + '/'.join(map(str, extra_arts)))) for artist in extra_arts: self.log.debug('Looking for artist similar ' 'to "{}" as well'.format(artist)) similar = self.ws_similar_artists(artist=artist) if not similar: return [] - ret_extra = set(self.get_artists_from_player(similar)) + ret_extra = self.get_artists_from_player(similar) if last_trk.Artist in ret_extra: ret_extra.remove(last_trk.Artist) return ret_extra @@ -222,12 +226,13 @@ class WebService(Plugin): return [] tolookfor = self.player.playlist[-1].Artist self.log.info('Looking for artist similar to "{}"'.format(tolookfor)) + self.log.debug(repr(tolookfor)) similar = self.ws_similar_artists(tolookfor) if not similar: self.log.info('Got nothing from {0}!'.format(self.ws.name)) return [] self.log.info('First five similar artist(s): {}...'.format( - ' / '.join([a.name for a in list(similar)[0:5]]))) + ' / '.join(map(str, list(similar)[:5])))) self.log.info('Looking availability in music library') ret = set(self.get_artists_from_player(similar)) ret_extra = None @@ -235,26 +240,34 @@ class WebService(Plugin): if self.plugin_conf.getint('depth') > 1: ret_extra = self.get_recursive_similar_artist() if ret_extra: - ret = set(ret) | set(ret_extra) + # get them reorg to pick up best element + ret_extra = self._get_artists_list_reorg(ret_extra) + # pickup half the number of ret artist + ret_extra = set(ret_extra[:len(ret)//2]) + self.log.debug('Using extra: {}'.format( + ' / '.join(map(str, ret_extra)))) + ret = ret | ret_extra if not ret: self.log.warning('Got nothing from music library.') self.log.warning('Try running in debug mode to guess why...') return [] - queued_artists = { trk.Artist for trk in self.player.queue } - for art in queued_artists: - if art in ret: - self.log.debug('Removing already queued artist: {0}'.format(art)) - ret = ret - queued_artists + # WARNING: + # * operation on set will not match against aliases + # * composite set w/ mbid set and whitout won't match either + queued_artists = {trk.Artist for trk in self.player.queue} if ret & queued_artists: - self.log.debug('Removing already queued artist: {0}'.format(ret & queued_artists)) + self.log.debug('Removing already queued artists: ' + '{0}'.format(ret & queued_artists)) ret = ret - queued_artists - if self.player.current.Artist in ret: + if self.player.current and self.player.current.Artist in ret: self.log.debug('Removing current artist: {0}'.format(self.player.current.Artist)) ret = ret - {self.player.current.Artist} # Move around similars items to get in unplayed|not recently played # artist first. self.log.info('Got {} artists in library'.format(len(ret))) - return self._get_artists_list_reorg(list(ret)) + candidates = self._get_artists_list_reorg(list(ret)) + self.log.info(' / '.join(map(str, candidates))) + return candidates def _get_album_history(self, artist=None): """Retrieve album history""" @@ -325,10 +338,8 @@ class WebService(Plugin): titles = [t for t in self.ws.get_toptrack(artist)] except WSError as err: self.log.warning('{0}: {1}'.format(self.ws.name, err)) - if self.ws.ratelimit: - self.log.info('{0.name} ratelimit: {0.ratelimit}'.format(self.ws)) for trk in titles: - found = self.player.fuzzy_find_track(artist.name, trk.title) + found = self.player.fuzzy_find_track(artist, trk.title) random.shuffle(found) if found: self.log.debug('{0}'.format(found[0])) @@ -384,7 +395,7 @@ class WebService(Plugin): self.queue_mode() msg = ' '.join(['{0}: {1:>3d}'.format(k, v) for k, v in sorted(self.ws.stats.items())]) - self.log.debug(msg) + self.log.debug('http stats: ' + msg) candidates = self.to_add self.to_add = list() if self.plugin_conf.get('queue_mode') != 'album':