Added support for unloading modules.

1 files changed, 21 insertions, 3 deletions

diff --git a/wmd/actions/modules.py b/wmd/actions/modules.py
index 60a1436..eb726c2 100644
--- a/wmd/actions/modules.py
+++ b/wmd/actions/modules.py
@@ -32,9 +32,9 @@ class LoadModule(Action):
 
         if username in settings.ADMINS:
             args = obj_data.params.split(" ")
-            if "PRIVMSG" in obj_data.command and "LOADMODULE" in args[1].upper():
+            if "PRIVMSG" in obj_data.command and ":LOADMODULE" in args[1].upper():
                 module_path = args[2]
-                if module_path in irc.actions:
+                if module_path.split(".")[-1] in irc.actions:
                     irc.privmsg(username, "Module %s already loaded" % (module_path,))
                     return
                 #irc.load_action(module_path)
@@ -50,7 +50,25 @@ class ListModules(Action):
 
         if username in settings.ADMINS:
             args = obj_data.params.split(" ")
-            if "PRIVMSG" in obj_data.command and "LISTMODULES" in args[1].upper():
+            if "PRIVMSG" in obj_data.command and ":LISTMODULES" in args[1].upper():
                 for module in irc.actions:
                     msg = module
                     irc.privmsg(username, msg)
+
+class UnloadModule(Action):
+    def recv_msg(self, irc, obj_data):
+        username = obj_data.get_username()
+
+        if username in settings.ADMINS:
+            args = obj_data.params.split(" ")
+            if "PRIVMSG" in obj_data.command and ":UNLOADMODULE" in args[1].upper():
+                module_name = args[2]
+                if not module_name in irc.actions:
+                    irc.privmsg(username, "Module %s already unloaded" % (module_name,))
+                    return
+                    #irc.load_action(module_path)
+
+                msg = "Unloading %s" % (module_name,)
+                self.log(msg)
+                irc.privmsg(username, msg)
+                return {'unload': module_name}
\ No newline at end of file