X-Git-Url: https://git.kaliko.me/?a=blobdiff_plain;f=sima%2Flib%2Fwebserv.py;h=0304dd639c1cd9866abb64026720e49a8ef3aec8;hb=751e07b3e43a4812c29f1c89f03e58023278b634;hp=0845924916f48f1316eea5946974182b033dd67b;hpb=d4e415f6ca4516f2403e582a05facda72db3c322;p=mpd-sima.git diff --git a/sima/lib/webserv.py b/sima/lib/webserv.py index 0845924..0304dd6 100644 --- a/sima/lib/webserv.py +++ b/sima/lib/webserv.py @@ -1,5 +1,6 @@ # -*- coding: utf-8 -*- -# Copyright (c) 2009-2015 Jack Kaliko +# Copyright (c) 2009-2019 Jack Kaliko +# Copyright (c) 2019 sacha # # This file is part of sima # @@ -32,7 +33,7 @@ from hashlib import md5 # local import from .plugin import Plugin from .track import Track -from .meta import Artist, MetaContainer +from .meta import Artist, Album, MetaContainer from ..utils.utils import WSError, WSNotFound def cache(func): @@ -112,17 +113,21 @@ class WebService(Plugin): * not blacklisted """ artist = tracks[0].artist - black_list = self.player.queue + self.to_add + # In random play mode use complete playlist to filter + if self.player.playmode.get('random'): + black_list = self.player.playlist + self.to_add + else: + black_list = self.player.queue + self.to_add not_in_hist = list(set(tracks) - set(self.get_history(artist=artist))) if self.plugin_conf.get('queue_mode') != 'top' and not not_in_hist: - self.log.debug('All tracks already played for "{}"'.format(artist)) + self.log.debug('All tracks already played for "%s"', artist) random.shuffle(not_in_hist) candidate = [] for trk in [_ for _ in not_in_hist if _ not in black_list]: # Should use albumartist heuristic as well if self.plugin_conf.getboolean('single_album'): # pylint: disable=no-member if (trk.album == self.player.current.album or - trk.album in [tr.album for tr in self.to_add]): + trk.album in [tr.album for tr in black_list]): self.log.debug('Found unplayed track ' + 'but from an album already queued: %s', trk) continue @@ -195,7 +200,11 @@ class WebService(Plugin): if not self.player.playlist: return history = list(self.history) - history = self.player.queue + history + # In random play mode use complete playlist to filter + if self.player.playmode.get('random'): + history = self.player.playlist + history + else: + history = self.player.queue + history history = deque(history) last_trk = history.popleft() # remove extra_arts = list() @@ -266,7 +275,11 @@ class WebService(Plugin): if not ret: self.log.warning('Got nothing from music library.') return [] - queued_artists = MetaContainer([trk.Artist for trk in self.player.queue]) + # In random play mode use complete playlist to filter + if self.player.playmode.get('random'): + queued_artists = MetaContainer([trk.Artist for trk in self.player.playlist]) + else: + queued_artists = MetaContainer([trk.Artist for trk in self.player.queue]) self.log.trace('Already queued: {}'.format(queued_artists)) self.log.trace('Candidate: {}'.format(ret)) if ret & queued_artists: @@ -312,25 +325,36 @@ class WebService(Plugin): albums_not_in_hist = list(albums - albums_yet_in_hist) # Get to next artist if there are no unplayed albums if not albums_not_in_hist: - self.log.info('No album found for "%s"' % artist) + self.log.info('No unplayed album found for "%s"' % artist) continue album_to_queue = str() random.shuffle(albums_not_in_hist) for album in albums_not_in_hist: - tracks = self.player.find_album(artist, album) - # Look if one track of the album is already queued - # Good heuristic, at least enough to guess if the whole album is - # already queued. - if tracks[0] in self.player.queue: - self.log.debug('"%s" already queued, skipping!', tracks[0].album) + # Controls the album found is not already queued + if album in {t.album for t in self.player.queue}: + self.log.debug('"%s" already queued, skipping!', album) continue + # In random play mode use complete playlist to filter + if self.player.playmode.get('random'): + if album in {t.album for t in self.player.playlist}: + self.log.debug('"%s" already in playlist, skipping!', album) + continue album_to_queue = album if not album_to_queue: self.log.info('No album found for "%s"', artist) continue self.log.info('%s album candidate: %s - %s', self.ws.name, artist, album_to_queue) nb_album_add += 1 - self.to_add.extend(self.player.find_album(artist, album_to_queue)) + candidates = self.player.find_tracks(Album(name=album_to_queue, + artist=artist)) + if self.plugin_conf.getboolean('shuffle_album'): + random.shuffle(candidates) + # this allows to select a maximum number of track from the album + # a value of 0 (default) means keep all + nbtracks = self.plugin_conf.getint('track_to_add_from_album') + if nbtracks > 0: + candidates = candidates[0:nbtracks] + self.to_add.extend(candidates) if nb_album_add == target_album_to_add: return True @@ -350,7 +374,7 @@ class WebService(Plugin): except WSError as err: self.log.warning('%s: %s', self.ws.name, err) for trk in titles: - found = self.player.fuzzy_find_track(artist, trk.title) + found = self.player.search_track(artist, trk.title) random.shuffle(found) if found: self.log.debug('%s', found[0])