Refactor Bot PluginManager configuration

attzonko · Aug 19, 2022 · 226ad94 · 226ad94
1 parent c44e072
commit 226ad94
Showing 1 changed file with 13 additions and 10 deletions.
diff --git a/mmpy_bot/bot.py b/mmpy_bot/bot.py
@@ -31,22 +31,15 @@ def __init__(
         plugins: Optional[Union[List[Plugin], PluginManager]] = None,
         enable_logging: bool = True,
     ):
-        if plugins is None:
-            self.plugin_manager = PluginManager(
-                [HelpPlugin(), ExamplePlugin(), WebHookExample()]
-            )
-        elif isinstance(plugins, PluginManager):
-            self.plugin_manager = plugins
-        else:
-            self.plugin_manager = PluginManager(plugins)
+        self._setup_plugin_manager(plugins)
 
         # Use default settings if none were specified.
         self.settings = settings or Settings()
 
+        self.console = None
+
         if enable_logging:
             self._register_logger()
-        else:
-            self.console = None
 
         self.driver = Driver(
             {
@@ -72,6 +65,16 @@ def __init__(
 
         self.running = False
 
+    def _setup_plugin_manager(self, plugins):
+        if plugins is None:
+            self.plugin_manager = PluginManager(
+                [HelpPlugin(), ExamplePlugin(), WebHookExample()]
+            )
+        elif isinstance(plugins, PluginManager):
+            self.plugin_manager = plugins
+        else:
+            self.plugin_manager = PluginManager(plugins)
+
     def _register_logger(self):
         logging.basicConfig(
             **{