Include some forgotten module in reload process

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)
 

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

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)

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()

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)