Proper loading of plugins on debug

Restarting functions

couchpotato/core/_base/_core/__init__.py

@@ -1,7 +1,8 @@
+from .main import Core
 from uuid import uuid4
 
 def start():
-    pass
+    return Core()
 
 config = [{
     'name': 'core',

couchpotato/core/_base/_core/main.py

@@ -0,0 +1,54 @@
+from couchpotato.api import addApiView
+from couchpotato.core.event import fireEvent
+from couchpotato.core.logger import CPLog
+from couchpotato.core.plugins.base import Plugin
+from couchpotato.environment import Env
+from flask import request
+import os
+
+
+log = CPLog(__name__)
+
+class Core(Plugin):
+
+    def __init__(self):
+        addApiView('app.shutdown', self.shutdown)
+        addApiView('app.restart', self.restart)
+
+        self.removeRestartFile()
+
+    def shutdown(self):
+        self.initShutdown()
+        return 'shutdown'
+
+    def restart(self):
+        self.initShutdown(restart = True)
+        return 'restarting'
+
+    def initShutdown(self, restart = False):
+
+        fireEvent('app.shutdown')
+
+        if restart:
+            self.writeRestartFile()
+
+        func = request.environ.get('werkzeug.server.shutdown')
+        if func is None:
+            raise RuntimeError('Not running with the Werkzeug Server')
+        func()
+
+    def removeRestartFile(self):
+        try:
+            os.remove(self.restartFilePath())
+        except:
+            pass
+
+    def writeRestartFile(self):
+        try:
+            with open(self.restartFilePath(), 'w') as f:
+                f.write('This is the most suckiest way to register if CP is restarted. Ever...')
+        except Exception, e:
+            log.error('Could not write shutdown file: %s' % e)
+
+    def restartFilePath(self):
+        return os.path.join(Env.get('data_dir'), 'restart')

couchpotato/core/_base/scheduler/main.py

@@ -20,6 +20,7 @@ class Scheduler(Plugin):
         addEvent('schedule.restart', self.start)
 
         addEvent('app.load', self.start)
+        addEvent('app.shutdown', self.stop)
 
         self.sched = Sched(misfire_grace_time = 60)