X-Git-Url: http://git.kaliko.me/?a=blobdiff_plain;f=sima%2Fplugins%2Finternal%2Fcrop.py;h=8f70acd7caf42d703bf75b6feeb339404dd37757;hb=ee218f16c2a449c6d72d550807114676e1e96d94;hp=8ea889e7810d08c7b88664c8b3bf86b87f087467;hpb=7854938788e0af521edd7199e40db796d3256351;p=mpd-sima.git

diff --git a/sima/plugins/internal/crop.py b/sima/plugins/internal/crop.py
index 8ea889e..8f70acd 100644
--- a/sima/plugins/internal/crop.py
+++ b/sima/plugins/internal/crop.py
@@ -1,27 +1,68 @@
 # -*- coding: utf-8 -*-
+# Copyright (c) 2013-2015, 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/>.
+#
+#
 """Crops playlist
 """
 
-# standart library import
-#from select import select
+# standard library import
 
-# third parties componants
+# third parties components
 
 # local import
 from ...lib.plugin import Plugin
 
+
 class Crop(Plugin):
     """
     Crop playlist on next track
-    kinda MPD's consume
     """
+    def __init__(self, daemon):
+        super().__init__(daemon)
+        self.daemon = daemon
+        self.target = None
+        if not self.plugin_conf:
+            return
+        target = self.plugin_conf.get('consume')
+        if not target:
+            return
+        try:
+            if int(target) < 0:
+                self.log.info('Negative value for consume, not cropping')
+                return
+        except ValueError:
+            self.log.warning('Bad value for consume, '
+                             'expecting an integer, not "%s"', target)
+        else:
+            self.target = int(target)
+            self.log.debug('Cropping at %s', self.target)
 
-    def callback_playlist(self):
-        player = self._Plugin__daemon.player
-        target_lengh = 10
-        while player.currentsong().pos > target_lengh:
+    def callback_next_song(self):
+        if not self.target:
+            return
+        if not self.daemon.enabled:
+            self.log.debug('Queueing disabled, not cropping')
+            return
+        player = self.daemon.player
+        if player.currentsong().pos > self.target:
             self.log.debug('cropping playlist')
-            player.remove()
+        while player.currentsong().pos > self.target:
+            player.delete(0)
 
 
 # VIM MODLINE