Improved filter_track plugin method

diff --git a/sima/lib/plugin.py b/sima/lib/plugin.py
index 1afafe5455297e5643fb0f977297bf5c3c0cf29c..db440133f4f7b99f027908b21e47f6a9f5adc2f8 100644 (file)
--- a/sima/lib/plugin.py
+++ b/sima/lib/plugin.py
@@ -23,8 +23,7 @@ Plugin object to derive from
 
 import random
 
-from .track import Track
-from .meta import Album, Artist, MetaContainer
+from .meta import Artist, MetaContainer
 
 
 class Plugin:
@@ -226,8 +225,10 @@ class AdvancedPlugin(Plugin):
         for trk in [_ for _ in not_in_hist if _ not in deny_list]:
             # Should use albumartist heuristic as well
             if self.plugin_conf.getboolean('single_album', False):  # pylint: disable=no-member
+                albums = [tr.album for tr in deny_list]
+                albums += [tr.album for tr in self.to_add]
                 if (trk.album == self.player.current.album or
-                        trk.album in [tr.album for tr in deny_list]):
+                        trk.album in albums):
                     self.log.debug('Found unplayed track ' +
                                    'but from an album already queued: %s', trk)
                     continue