Merge pull request #340 from gunthercox/config

Allow chat bot to be created from json config file

chatterbot/chatterbot.py

@@ -235,6 +235,16 @@ def train(self):
         """
         return self.trainer.train
 
+    @classmethod
+    def from_config(self, config_file_path):
+        import json
+        with open(config_file_path, 'r') as config_file:
+            data = json.load(config_file)
+
+        name = data.pop('name')
+
+        return ChatBot(name, **data)
+
     class InvalidAdapterException(Exception):
 
         def __init__(self, value='Recieved an unexpected adapter setting.'):

tests/base_case.py

@@ -72,5 +72,4 @@ def get_kwargs(self):
         kwargs = super(ChatBotMongoTestCase, self).get_kwargs()
         kwargs['database'] = self.random_string()
         kwargs['storage_adapter'] = 'chatterbot.adapters.storage.MongoDatabaseAdapter'
-        return kwargs
-
+        return kwargs

tests/test_chatbot.py

@@ -148,3 +148,29 @@ def test_response_extra_data(self):
 
         self.assertIn("test", saved_statement.extra_data)
         self.assertEqual(1, saved_statement.extra_data["test"])
+
+
+class ChatBotConfigFileTestCase(ChatBotTestCase):
+
+    def setUp(self):
+        super(ChatBotConfigFileTestCase, self).setUp()
+        import json
+        self.config_file_path = './test-config.json'
+        self.data = self.get_kwargs()
+        self.data['name'] = 'Config Test'
+
+        with open(self.config_file_path, 'w+') as config_file:
+            json.dump(self.data, config_file)
+
+    def tearDown(self):
+        super(ChatBotConfigFileTestCase, self).tearDown()
+        import os
+
+        if os.path.exists(self.config_file_path):
+            os.remove(self.config_file_path)
+
+    def test_read_from_config_file(self):
+        from chatterbot import ChatBot
+        self.chatbot = ChatBot.from_config(self.config_file_path)
+
+        self.assertEqual(self.chatbot.name, self.data['name'])