Fix and improve reload process

nemubot/__init__.py

@@ -89,6 +89,9 @@ def reload():
 
     import imp
 
+    import nemubot.bot
+    imp.reload(nemubot.bot)
+
     import nemubot.channel
     imp.reload(nemubot.channel)
 

nemubot/__main__.py

@@ -141,7 +141,8 @@ def main():
 
     # Register the hook for futur import
     from nemubot.importer import ModuleFinder
-    sys.meta_path.append(ModuleFinder(context.modules_paths, context.add_module))
+    module_finder = ModuleFinder(context.modules_paths, context.add_module)
+    sys.meta_path.append(module_finder)
 
     # Load requested configuration files
     for path in args.files:
@@ -164,6 +165,9 @@ def main():
     def sighuphandler(signum, frame):
         """On SIGHUP, perform a deep reload"""
         import imp
+        nonlocal nemubot, context, module_finder
+
+        logger.debug("SIGHUP receive, iniate reload procedure...")
 
         # Reload nemubot Python modules
         imp.reload(nemubot)
@@ -173,6 +177,11 @@ def main():
         import nemubot.bot
         context = nemubot.bot.hotswap(context)
 
+        # Reload ModuleFinder
+        sys.meta_path.remove(module_finder)
+        module_finder = ModuleFinder(context.modules_paths, context.add_module)
+        sys.meta_path.append(module_finder)
+
         # Reload configuration file
         for path in args.files:
             if os.path.isfile(path):
@@ -188,9 +197,6 @@ def main():
                          "".join(traceback.format_stack(stack)))
     signal.signal(signal.SIGUSR1, sigusr1handler)
 
-    # Here we go!
-    context.start()
-
     if args.socketfile:
         from nemubot.server.socket import SocketListener
         context.add_server(SocketListener(context.add_server, "master_socket",
@@ -200,6 +206,7 @@ def main():
     oldcontext = None
     while oldcontext != context:
         oldcontext = context
+        context.start()
         context.join()
 
     # Wait for consumers

nemubot/bot.py

@@ -143,6 +143,7 @@ class Bot(threading.Thread):
         from select import select
         from nemubot.server import _lock, _rlist, _wlist, _xlist
 
+        logger.info("Starting main loop")
         self.stop = False
         while not self.stop:
             with _lock:
@@ -208,6 +209,7 @@ class Bot(threading.Thread):
                         from nemubot.tools.config import load_file
                         load_file(path, self)
                 self.sync_queue.task_done()
+        logger.info("Ending main loop")
 
 
 
@@ -509,14 +511,16 @@ def hotswap(bak):
     bak.stop = True
     if bak.event_timer is not None:
         bak.event_timer.cancel()
+
+    # Unload modules
+    for mod in [k for k in bak.modules.keys()]:
+        bak.unload_module(mod)
+
+    # Save datastore
     bak.datastore.close()
 
     new = Bot(str(bak.ip), bak.modules_paths, bak.datastore)
     new.servers = bak.servers
-    new.modules = bak.modules
-    new.modules_configuration = bak.modules_configuration
-    new.events = bak.events
-    new.hooks = bak.hooks
 
     new._update_event_timer()
     return new

nemubot/server/abstract.py

@@ -115,7 +115,7 @@ class AbstractServer(io.IOBase):
         """
 
         self._sending_queue.put(self.format(message))
-        self.logger.debug("Message '%s' appended to Queue", message)
+        self.logger.debug("Message '%s' appended to write queue", message)
         if self not in _wlist:
             _wlist.append(self)