Fix markov import tool to use the new plugin API

Signed-off-by: Alek Ratzloff <alekratz@gmail.com>
This commit is contained in:
2022-05-30 17:16:22 -07:00
parent dd4b7f2223
commit 1b4084322b

View File

@@ -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<name>[^>]+)> (?P<message>.+)$")