intercal/omnibot22
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix markov import tool to use the new plugin API

Browse Source 
Signed-off-by: Alek Ratzloff <alekratz@gmail.com>
This commit is contained in:
Alek Ratzloff
2022-05-30 17:16:22 -07:00
parent dd4b7f2223
commit 1b4084322b
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tools/markov_import.py
View File

@@ -4,6 +4,7 @@ import sys
import re import re
from omnibot.config import ServerConfig from omnibot.config import ServerConfig
from omnibot.bot import Bot
from plugins.markov import Markov from plugins.markov import Markov
@@ -24,10 +25,8 @@ async def main():
server_config = ServerConfig() server_config = ServerConfig()
server_config.load("config.toml") server_config.load("config.toml")
# This only works on one plugin per config # This only works on one plugin per config
plugin_config = [ bot = Bot(server_config)
plugin for plugin in server_config.plugins if "markov" in plugin["module"] plugin = [plugin for plugin in bot.plugins if isinstance(plugin, Markov)][0]
][0]
plugin = Markov(server_config, plugin_config)
LINE_RE = re.compile(r"^\[[^\]]+\] <(?P<name>[^>]+)> (?P<message>.+)$") LINE_RE = re.compile(r"^\[[^\]]+\] <(?P<name>[^>]+)> (?P<message>.+)$")