Scheduler

NZBmatrix, newznab
Scores

couchpotato/core/_base/_core/__init__.py

@@ -0,0 +1,74 @@
+from uuid import uuid4
+
+def start():
+    pass
+
+config = [{
+    'name': 'core',
+    'groups': [
+        {
+            'tab': 'general',
+            'name': 'basics',
+            'description': 'Needs restart before changes take effect.',
+            'options': [
+                {
+                    'name': 'username',
+                    'default': '',
+                },
+                {
+                    'name': 'password',
+                    'default': '',
+                    'type': 'password',
+                },
+                {
+                    'name': 'host',
+                    'advanced': True,
+                    'default': '0.0.0.0',
+                    'label': 'IP',
+                    'description': 'Host that I should listen to. "0.0.0.0" listens to all ips.',
+                },
+                {
+                    'name': 'port',
+                    'default': 5000,
+                    'type': 'int',
+                    'description': 'The port I should listen to.',
+                },
+                {
+                    'name': 'launch_browser',
+                    'default': 1,
+                    'type': 'bool',
+                    'label': 'Launch Browser',
+                    'description': 'Launch the browser when I start.',
+                },
+            ],
+        },
+        {
+            'tab': 'general',
+            'name': 'advanced',
+            'description': "For those who know what the're doing",
+            'advanced': True,
+            'options': [
+                {
+                    'name': 'api_key',
+                    'default': uuid4().hex,
+                    'readonly': 1,
+                    'label': 'Api Key',
+                    'description': "This is top-secret! Don't share this!",
+                },
+                {
+                    'name': 'debug',
+                    'default': 0,
+                    'type': 'bool',
+                    'label': 'Debug',
+                    'description': 'Enable debugging.',
+                },
+                {
+                    'name': 'url_base',
+                    'default': '',
+                    'label': 'Url Base',
+                    'description': 'When using mod_proxy use this to append the url with this.',
+                },
+            ],
+        },
+    ],
+}]

couchpotato/core/_base/clientscript/__init__.py

@@ -0,0 +1,6 @@
+from .main import ClientScript
+
+def start():
+    return ClientScript()
+
+config = []

couchpotato/core/_base/clientscript/main.py

@@ -0,0 +1,56 @@
+from couchpotato.core.event import addEvent
+from couchpotato.core.logger import CPLog
+from couchpotato.core.plugins.base import Plugin
+
+log = CPLog(__name__)
+
+
+class ClientScript(Plugin):
+
+    urls = {
+        'style': {},
+        'script': {},
+    }
+
+    html = {
+        'style': '<link rel="stylesheet" href="%s" type="text/css">',
+        'script': '<script type="text/javascript" src="%s"></script>',
+    }
+
+    def __init__(self):
+        addEvent('register_style', self.registerStyle)
+        addEvent('register_script', self.registerScript)
+
+        addEvent('clientscript.get_styles', self.getStyles)
+        addEvent('clientscript.get_scripts', self.getScripts)
+
+    def getStyles(self, *args, **kwargs):
+        return self.get('style', *args, **kwargs)
+
+    def getScripts(self, *args, **kwargs):
+        return self.get('script', *args, **kwargs)
+
+    def get(self, type, as_html = False, location = 'head'):
+
+        data = '' if as_html else []
+
+        try:
+            return self.urls[type][location]
+        except Exception, e:
+            log.error(e)
+
+        return data
+
+    def registerStyle(self, path, position = 'head'):
+        self.register(path, 'style', position)
+
+    def registerScript(self, path, position = 'head'):
+        self.register(path, 'script', position)
+
+    def register(self, file, type, location):
+
+        if not self.urls[type].get(location):
+            self.urls[type][location] = []
+
+        filePath = file
+        self.urls[type][location].append(filePath)

couchpotato/core/_base/scheduler/__init__.py

@@ -0,0 +1,6 @@
+from .main import Scheduler
+
+def start():
+    return Scheduler()
+
+config = []

couchpotato/core/_base/scheduler/main.py

@@ -0,0 +1,85 @@
+from apscheduler.scheduler import Scheduler as Sched
+from couchpotato.core.event import addEvent
+from couchpotato.core.logger import CPLog
+from couchpotato.core.plugins.base import Plugin
+
+log = CPLog(__name__)
+
+
+class Scheduler(Plugin):
+
+    crons = {}
+    intervals = {}
+    started = False
+
+    def __init__(self):
+
+        addEvent('schedule.cron', self.cron)
+        addEvent('schedule.interval', self.interval)
+        addEvent('schedule.start', self.start)
+        addEvent('schedule.restart', self.start)
+
+        addEvent('app.load', self.start)
+
+        self.sched = Sched(misfire_grace_time = 60)
+
+    def remove(self, identifier):
+        for type in ['interval', 'cron']:
+            try:
+                self.sched.unschedule_job(getattr(self, type)[identifier]['job'])
+                log.debug('%s unscheduled %s' % (type.capitalize(), identifier))
+            except:
+                pass
+
+    def start(self):
+
+        # Stop all running
+        self.stop()
+
+        # Crons
+        for identifier in self.crons:
+            self.remove(identifier)
+            cron = self.crons[identifier]
+            job = self.sched.add_cron_job(cron['handle'], day = cron['day'], hour = cron['hour'], minute = cron['minute'])
+            cron['job'] = job
+
+        # Intervals
+        for identifier in self.intervals:
+            self.remove(identifier)
+            interval = self.intervals[identifier]
+            job = self.sched.add_interval_job(interval['handle'], hours = interval['hours'], minutes = interval['minutes'], seconds = interval['seconds'], repeat = interval['repeat'])
+            interval['job'] = job
+
+        # Start it
+        self.sched.start()
+        self.started = True
+
+    def stop(self):
+
+        if self.started:
+            self.sched.shutdown()
+
+        self.started = False
+
+    def cron(self, identifier = '', handle = None, day = '*', hour = '*', minute = '*'):
+        log.info('Scheduling "%s", cron: day = %s, hour = %s, minute = %s' % (identifier, day, hour, minute))
+
+        self.remove(identifier)
+        self.crons[identifier] = {
+            'handle': handle,
+            'day': day,
+            'hour': hour,
+            'minute': minute,
+        }
+
+    def interval(self, identifier = '', handle = None, hours = 0, minutes = 0, seconds = 0, repeat = 0):
+        log.info('Scheduling %s, interval: hours = %s, minutes = %s, seconds = %s, repeat = %s' % (identifier, hours, minutes, seconds, repeat))
+
+        self.remove(identifier)
+        self.intervals[identifier] = {
+            'handle': handle,
+            'repeat': repeat,
+            'hours': hours,
+            'minutes': minutes,
+            'seconds': seconds,
+        }