Switched event system (again)

2011-03-01 11:12:56 +01:00
parent fd5a41435b
commit f9523444f4
24 changed files with 580 additions and 142 deletions
@@ -0,0 +1,86 @@
+from couchpotato.core.event import fireEvent, fireEventAsync
+from couchpotato.core.logger import CPLog
+import glob
+import os
+
+log = CPLog(__name__)
+
+class Loader:
+
+    plugins = {}
+    providers = {}
+
+    modules = {}
+
+    def preload(self, root = ''):
+
+        self.paths = {
+            'plugin' : ('couchpotato.core.plugins', os.path.join(root, 'couchpotato', 'core', 'plugins')),
+            'provider' : ('couchpotato.core.providers', os.path.join(root, 'couchpotato', 'core', 'providers')),
+        }
+
+        for type, tuple in self.paths.iteritems():
+            self.addFromDir(type, tuple[0], tuple[1])
+
+    def run(self):
+        did_save = 0
+
+        for module_name, plugin in sorted(self.modules.iteritems()):
+
+            # Load module
+            m = getattr(self.loadModule(module_name), plugin.get('name'))
+
+            log.info("Loading '%s'" % module_name)
+
+            # Save default settings for plugin/provider
+            did_save += self.loadSettings(m, module_name, save = False)
+
+            self.loadPlugins(m, plugin.get('name'))
+
+        if did_save:
+            fireEvent('settings.save')
+
+    def addFromDir(self, type, module, dir):
+
+        for file in glob.glob(os.path.join(dir, '*')):
+            name = os.path.basename(file)
+            if os.path.isdir(os.path.join(dir, name)):
+                module_name = '%s.%s' % (module, name)
+                self.addModule(type, module_name, name)
+
+    def loadSettings(self, module, name, save = True):
+        try:
+            for section in module.config:
+                fireEventAsync('settings.options', section['name'], section)
+                options = {}
+                for key, option in section['options'].iteritems():
+                    options[key] = option['default']
+                fireEventAsync('settings.register', section_name = section['name'], options = options, save = save)
+            return True
+        except Exception, e:
+            log.debug("Failed loading settings for '%s': %s" % (name, e))
+            return False
+
+    def loadPlugins(self, module, name):
+        try:
+            module.start()
+            return True
+        except Exception, e:
+            log.debug("Failed loading plugin '%s': %s" % (name, e))
+            return False
+
+    def addModule(self, type, module, name):
+        self.modules[module] = {
+            'type': type,
+            'name': name,
+        }
+
+    def loadModule(self, name):
+        try:
+            m = __import__(name)
+            splitted = name.split('.')
+            for sub in splitted[1:-1]:
+                m = getattr(m, sub)
+            return m
+        except:
+            raise