Improved module handler; improved mensa formatting

1 files changed, 49 insertions, 0 deletions

diff --git a/modules/__init__.py b/modules/__init__.py
index 49a460a..403bbc9 100644
--- a/modules/__init__.py
+++ b/modules/__init__.py
@@ -19,3 +19,52 @@ class ModuleBase:
     
     def topic(self, room, nick, text, handler):
         pass
+
+class ModuleManager:
+    def __init__(self, config):
+        self.config = config
+        self._modules = {}
+        
+        for mod in config['modules']:
+            self.get(mod)
+    
+    def get(self, name):
+        if not name in self._modules:
+            mod = __import__('modules.' + name, globals(), locals(), ['Module'])
+            self._modules[name] = mod.Module(self)
+        
+        return self._modules[name]
+    
+    def commands(self):
+        return reduce(lambda l, mod: l + mod.commands(), self._modules.itervalues(), [])
+    
+    def helptexts(self):
+        return reduce(lambda l, mod: l + mod.helptexts(), self._modules.itervalues(), [])
+    
+    def groupchat(self, room, nick, text, handler):
+        for mod in self._modules.itervalues():
+            try:
+                mod.groupchat(room, nick, text, handler)
+            except:
+                handler.reply(traceback.format_exc(5))
+    
+    def join(self, room, nick, show, status, handler):
+        for mod in self._modules.itervalues():
+            try:
+                mod.join(room, nick, show, status, handler)
+            except:
+                handler.reply(traceback.format_exc(5))
+    
+    def leave(self, room, nick, show, status, handler):
+        for mod in self._modules.itervalues():
+            try:
+                mod.leave(room, nick, show, status, handler)
+            except:
+                handler.reply(traceback.format_exc(5))
+    
+    def topic(self, room, nick, text, handler):
+        for mod in self._modules.itervalues():
+            try:
+                mod.topic(room, nick, text, handler)
+            except:
+                handler.reply(traceback.format_exc(5))