1 files changed, 78 insertions, 0 deletions
diff --git a/module/plugins/hooks/DownloadScheduler.py b/module/plugins/hooks/DownloadScheduler.py
new file mode 100644
index 000000000..2d7d2ffb0
--- /dev/null
+++ b/module/plugins/hooks/DownloadScheduler.py
@@ -0,0 +1,78 @@
+# -*- coding: utf-8 -*-
+
+"""
+    This program 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.
+
+    This program 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.
+    You should have received a copy of the GNU General Public License
+    along with this program; if not, see <http://www.gnu.org/licenses/>.
+    
+    @author: zoidberg
+    Original idea by new.cze
+"""
+
+import re
+from time import localtime
+from module.plugins.Hook import Hook
+
+class DownloadScheduler(Hook):
+    __name__ = "DownloadScheduler"
+    __version__ = "0.20"
+    __description__ = """Download Scheduler"""
+    __config__ = [("activated", "bool", "Activated", "False"),                  
+                  ("timetable", "str", "List time periods as hh:mm full or number(kB/s)", "0:00 full, 7:00 250, 10:00 0, 17:00 150")]
+    __author_name__ = ("zoidberg")
+    __author_mail__ = ("zoidberg@mujmail.cz")
+    
+    def setup(self):
+        self.cb = None # callback to scheduler job; will be by removed hookmanager when hook unloaded       
+    
+    def coreReady(self):
+        self.updateSchedule()
+        
+    def updateSchedule(self, schedule = None):        
+        if schedule is None: 
+            schedule = self.getConfig("timetable")                   
+        
+        schedule = re.findall("(\d{1,2}):(\d{2})[\s]*(-?\d+)", schedule.lower().replace("full", "-1").replace("none", "0"))         
+        if not schedule:
+            self.logError("Invalid schedule")
+            return
+        
+        t0 = localtime()
+        now = (t0.tm_hour, t0.tm_min, t0.tm_sec, "X")
+        schedule = sorted([(int(x[0]), int(x[1]), 0, int(x[2])) for x in schedule] + [now])
+    
+        self.logDebug("Schedule", schedule)      
+      
+        for i, v in enumerate(schedule):
+            if v[3] == "X":
+                last, next = schedule[i-1], schedule[(i+1) % len(schedule)]
+                self.logDebug("Now/Last/Next", now, last, next)
+                 
+                self.setDownloadSpeed(last[3]) 
+                
+                next_time = (((24 + next[0] - now[0])* 60 + next[1] - now[1]) * 60 + next[2] - now[2]) % 86400
+                self.core.scheduler.removeJob(self.cb)
+                self.cb = self.core.scheduler.addJob(next_time, self.updateSchedule, threaded=False)                                     
+                
+    def setDownloadSpeed(self, speed):     
+        if speed == 0:
+            self.logInfo("Stopping download server. Current downloads will not be interrupted.")
+            self.core.api.pauseServer()
+        else:
+            self.core.api.unpauseServer()
+            
+            if speed > 0:
+                self.logInfo("Setting download speed to %d kB/s" % speed)
+                self.core.api.setConfigValue("download","limit_speed",1)
+                self.core.api.setConfigValue("download","max_speed",speed)
+            else:
+                self.logInfo("Setting download speed to FULL")
+                self.core.api.setConfigValue("download","limit_speed",0)
+                self.core.api.setConfigValue("download","max_speed",-1)
+\ No newline at end of file