From: kaliko <kaliko@azylum.org>
Date: Thu, 7 May 2020 14:28:21 +0000 (+0200)
Subject: Start work on new tags plugin
X-Git-Tag: 0.16.0~16
X-Git-Url: https://git.kaliko.me/?a=commitdiff_plain;h=29cca9ed73ed4bd87b909398ccc80bceddc89c73;p=mpd-sima.git

Start work on new tags plugin
---

diff --git a/sima/plugins/internal/tags.py b/sima/plugins/internal/tags.py
new file mode 100644
index 0000000..cd1a4fe
--- /dev/null
+++ b/sima/plugins/internal/tags.py
@@ -0,0 +1,117 @@
+# -*- coding: utf-8 -*-
+# Copyright (c) 2020 kaliko <kaliko@azylum.org>
+#
+#  This file is part of sima
+#
+#  sima is free software: you can redistribute it and/or modify
+#  it under the terms of the GNU General Public License as published by
+#  the Free Software Foundation, either version 3 of the License, or
+#  (at your option) any later version.
+#
+#  sima is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU General Public License for more details.
+#
+#  You should have received a copy of the GNU General Public License
+#  along with sima.  If not, see <http://www.gnu.org/licenses/>.
+#
+#
+"""
+Add titles based on tags
+"""
+
+# standard library import
+import random
+
+# third parties components
+
+# local import
+from ...lib.plugin import Plugin
+from ...lib.track import Track
+from ...utils.utils import PluginConfException
+
+def forge_filter(cfg):
+    tags = set(cfg.keys()) & Tags.supported_tags
+    cfg_filter = cfg.get('filter', None)
+    mpd_filter = []
+    if cfg_filter:
+        mpd_filter.append(cfg_filter)
+    for tag in tags:
+        if ',' in cfg[tag]:
+            patt = '|'.join(cfg[tag].split(','))
+            mpd_filter.append(f"({tag} =~ '({patt})')")
+        else:
+            mpd_filter.append(f"({tag} == '{cfg[tag].strip()}')")
+    mpd_filter = ' AND '.join(mpd_filter)
+    if 'AND' in mpd_filter:
+        mpd_filter = f'({mpd_filter})'
+    return mpd_filter
+
+
+class Tags(Plugin):
+    """Add track based on tags content
+    """
+    supported_tags = {'comment', 'date', 'genre', 'label', 'originaldate'}
+
+    def __init__(self, daemon):
+        super().__init__(daemon)
+        self.daemon = daemon
+        self._control_conf()
+        #self._control_server()
+        self._setup_tagsneeded()
+        self.mpd_filter = forge_filter(self.plugin_conf)
+        self.log.debug('mpd filter: %s', self.mpd_filter)
+
+    def _control_conf(self):
+        sup_tags = Tags.supported_tags
+        if not self.plugin_conf.get('filter', None) and \
+                self.plugin_conf.keys().isdisjoint(sup_tags):
+            self.log.error(
+                'Found no config for %s plugin! Need at least "filter" or a supported tag' % self)
+            self.log.info('Supported Tags are : %s', ', '.join(sup_tags))
+            raise PluginConfException('plugin misconfiguration')
+
+    def _control_server(self):
+        #TODO:
+        # * control tags used are available
+        # * filters are available mpd version >= 0.21
+        raise NotImplemented
+
+    def _setup_tagsneeded(self):
+        tags = set(self.plugin_conf.keys()) & Tags.supported_tags
+        self.player.needed_tags |= tags
+
+    def _get_history(self):
+        """Constructs list of already played artists.
+        """
+        duration = self.daemon.config.getint('sima', 'history_duration')
+        tracks_from_db = self.daemon.sdb.get_history(duration=duration)
+        hist = [Track(file=tr[3], artist=tr[0]) for tr in tracks_from_db]
+        return hist
+
+    def callback_need_track(self):
+        candidates = []
+        target = self.plugin_conf.getint('track_to_add')
+        tracks = self.player.find(self.mpd_filter)
+        random.shuffle(tracks)
+        history = self._get_history()
+        while tracks:
+            trk = tracks.pop()
+            if trk in self.player.queue or \
+               trk in candidates:
+                self.log.debug('%s already queued', trk)
+                continue
+            if trk in history:
+                self.log.debug('%s in history', trk)
+                continue
+            candidates.append(trk)
+            self.log.info('Tags candidate: {}'.format(trk))
+            if len(candidates) >= target:
+                break
+        if not candidates:
+            self.log.info('Tags plugin failed to find some tracks')
+        return candidates
+
+# VIM MODLINE
+# vim: ai ts=4 sw=4 sts=4 expandtab