From 1b4084322b0b5b3b45b482878ed2e7b933972668 Mon Sep 17 00:00:00 2001 From: Alek Ratzloff Date: Mon, 30 May 2022 17:16:22 -0700 Subject: [PATCH] Fix markov import tool to use the new plugin API Signed-off-by: Alek Ratzloff --- tools/markov_import.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tools/markov_import.py b/tools/markov_import.py index 5270f00..490f47a 100644 --- a/tools/markov_import.py +++ b/tools/markov_import.py @@ -4,6 +4,7 @@ import sys import re from omnibot.config import ServerConfig +from omnibot.bot import Bot from plugins.markov import Markov @@ -24,10 +25,8 @@ async def main(): server_config = ServerConfig() server_config.load("config.toml") # This only works on one plugin per config - plugin_config = [ - plugin for plugin in server_config.plugins if "markov" in plugin["module"] - ][0] - plugin = Markov(server_config, plugin_config) + bot = Bot(server_config) + plugin = [plugin for plugin in bot.plugins if isinstance(plugin, Markov)][0] LINE_RE = re.compile(r"^\[[^\]]+\] <(?P[^>]+)> (?P.+)$")