From c6aa38147b8d55843f6770eaed66d69ed9a07402 Mon Sep 17 00:00:00 2001
From: nemunaire <nemunaire@nemunai.re>
Date: Fri, 30 Oct 2015 22:18:48 +0100
Subject: [PATCH] Include some forgotten module in reload process

---
 nemubot/__init__.py                 |  5 +++++
 nemubot/datastore/__init__.py       | 13 +++++++++++++
 nemubot/message/printer/__init__.py |  3 +++
 nemubot/server/__init__.py          |  5 +++++
 nemubot/tools/__init__.py           |  3 +++
 5 files changed, 29 insertions(+)

diff --git a/nemubot/__init__.py b/nemubot/__init__.py
index 044d993..005e2b1 100644
--- a/nemubot/__init__.py
+++ b/nemubot/__init__.py
@@ -50,6 +50,11 @@ def reload():
     import nemubot.consumer
     imp.reload(nemubot.consumer)
 
+    import nemubot.datastore
+    imp.reload(nemubot.datastore)
+
+    nemubot.datastore.reload()
+
     import nemubot.event
     imp.reload(nemubot.event)
 
diff --git a/nemubot/datastore/__init__.py b/nemubot/datastore/__init__.py
index 323a160..ed9e829 100644
--- a/nemubot/datastore/__init__.py
+++ b/nemubot/datastore/__init__.py
@@ -16,3 +16,16 @@
 
 from nemubot.datastore.abstract import Abstract
 from nemubot.datastore.xml import XML
+
+
+def reload():
+    global Abstract, XML
+    import imp
+
+    import nemubot.datastore.abstract
+    imp.reload(nemubot.datastore.abstract)
+    Abstract = nemubot.datastore.abstract.Abstract
+
+    import nemubot.datastore.xml
+    imp.reload(nemubot.datastore.xml)
+    XML = nemubot.datastore.xml.XML
diff --git a/nemubot/message/printer/__init__.py b/nemubot/message/printer/__init__.py
index ae6b4df..bb58338 100644
--- a/nemubot/message/printer/__init__.py
+++ b/nemubot/message/printer/__init__.py
@@ -19,3 +19,6 @@ def reload():
 
     import nemubot.message.printer.IRC
     imp.reload(nemubot.message.printer.IRC)
+
+    import nemubot.message.printer.socket
+    imp.reload(nemubot.message.printer.socket)
diff --git a/nemubot/server/__init__.py b/nemubot/server/__init__.py
index 1f68d74..b9a8fe4 100644
--- a/nemubot/server/__init__.py
+++ b/nemubot/server/__init__.py
@@ -81,3 +81,8 @@ def reload():
 
     import nemubot.server.IRC
     imp.reload(nemubot.server.IRC)
+
+    import nemubot.server.message
+    imp.reload(nemubot.server.message)
+
+    nemubot.server.message.reload()
diff --git a/nemubot/tools/__init__.py b/nemubot/tools/__init__.py
index 9043466..127154c 100644
--- a/nemubot/tools/__init__.py
+++ b/nemubot/tools/__init__.py
@@ -23,6 +23,9 @@ def reload():
     import nemubot.tools.countdown
     imp.reload(nemubot.tools.countdown)
 
+    import nemubot.tools.feed
+    imp.reload(nemubot.tools.feed)
+
     import nemubot.tools.date
     imp.reload(nemubot.tools.date)