fixed a bug that crashed warmachine if you tried loading a class that doesn't exist in a module that does.

1 files changed, 6 insertions, 1 deletions

diff --git a/wmd/irc.py b/wmd/irc.py
index 813f743..b06cfec 100644
--- a/wmd/irc.py
+++ b/wmd/irc.py
@@ -70,7 +70,12 @@ class IRC(object):
             self.log("Error loading module: %s" %(path,))
             return
 
-        classz = getattr(module, class_name)
+        try:
+            classz = getattr(module, class_name)
+        except AttributeError:
+            self.log("Class does not exist in module")
+            return
+
         self.actions[class_name] = classz()
 
     def __call__(self, *args, **kwargs):