From 052e741da03264d4312650618e831abfd4e66f84 Mon Sep 17 00:00:00 2001
From: jesopo
Date: Thu, 14 Apr 2016 16:47:33 +0100
Subject: added a very unreliable and BAD IDEA way to unload modules.

---
 EventManager.py  |  2 ++
 ModuleManager.py | 26 +++++++++++++++++++++++---
 2 files changed, 25 insertions(+), 3 deletions(-)

diff --git a/EventManager.py b/EventManager.py
index 3b75de9f..e17258df 100644
--- a/EventManager.py
+++ b/EventManager.py
@@ -46,6 +46,8 @@ class EventHook(object):
         if self._hook_notify:
             self._hook_notify(self, callback)
         self._hooks.append(callback)
+    def _unhook(self, hook):
+        self._hooks.remove(hook)
     def on(self, subevent, *extra_subevents):
         if extra_subevents:
             multiple_event_hook = MultipleEventHook()
diff --git a/ModuleManager.py b/ModuleManager.py
index fc30711e..176171f7 100644
--- a/ModuleManager.py
+++ b/ModuleManager.py
@@ -1,4 +1,4 @@
-import glob, imp, inspect, os, sys
+import gc, glob, imp, inspect, os, sys
 
 class ModuleManager(object):
     def __init__(self, bot, directory="modules"):
@@ -38,7 +38,8 @@ class ModuleManager(object):
                             return None
                 else:
                     break
-        module = imp.load_source("bitbot_%s" % name, filename)
+        import_name = "bitbot_%s" % name
+        module = imp.load_source(import_name, filename)
         assert hasattr(module, "Module"
             ), "module '%s' doesn't have a Module class."
         assert inspect.isclass(module.Module
@@ -46,6 +47,8 @@ class ModuleManager(object):
         module_object = module.Module(self.bot)
         if not hasattr(module_object, "_name"):
             module_object._name = name.title()
+        module_object._is_unloaded = False
+        module_object._import_name = import_name
         assert not module_object._name in self.modules, (
             "module name '%s' attempted to be used twice.")
         return module_object
@@ -60,7 +63,24 @@ class ModuleManager(object):
                     self.load_module(filename)
         else:
             sys.stderr.write("module '%s' not loaded.\n" % filename)
-
     def load_modules(self):
         for filename in self.list_modules():
             self.load_module(filename)
+
+    def unload_module(self, module):
+        # this is such a bad idea
+        module._is_unloaded = True
+        self.unhook_check(self.bot.events)
+        del sys.modules[module._import_name]
+        del self.modules[module._name]
+        del module
+        gc.collect()
+
+    def unhook_check(self, event):
+        for hook in event.get_hooks():
+            if hasattr(hook.function, "__self__") and hasattr(
+                    hook.function.__self__, "_is_unloaded"
+                    ) and hook.function.__self__._is_unloaded:
+                event._unhook(hook)
+        for child in event.get_children():
+            self.unhook_check(event.get_child(child))
-- 
cgit v1.3.1-10-gc9f91