From 465e256d17cb5c43972ddc0af660d197af38b501 Mon Sep 17 00:00:00 2001 From: deathbybandaid Date: Fri, 7 Jun 2019 08:35:15 -0400 Subject: [PATCH] test --- sopel_modules/SpiceBot/AI.py | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/sopel_modules/SpiceBot/AI.py b/sopel_modules/SpiceBot/AI.py index 23d3919..315d148 100644 --- a/sopel_modules/SpiceBot/AI.py +++ b/sopel_modules/SpiceBot/AI.py @@ -14,6 +14,8 @@ import aiml from .Database import db as botdb from .Config import config as botconfig +import spicemanip + class SpiceBot_AI_MainSection(StaticSection): extra = ListAttribute('extra') @@ -137,9 +139,6 @@ class SpiceBot_AI(): def bot_message_precipher(self, bot, trigger, message): - # uppercase input - message = message.upper() - # punctuation puct_dict = {"!": "exclamationmark", ".": "period", "?": "questionmark", ",": "comma"} for puctuation in puct_dict.keys(): @@ -147,10 +146,18 @@ class SpiceBot_AI(): # bot items for botitem in ["nick"]: - message = message.replace(str(eval("bot." + botitem)).upper(), str("bot" + botitem).upper()) + messagelist = spicemanip.main(message, "create") + for i in range(len(messagelist)): + if messagelist[i].upper() == str(eval("bot." + botitem)).upper(): + messagelist[i] = str("bot" + botitem).upper() + message = spicemanip.main(messagelist, 0) for triggeritem in ["nick", "sender"]: - message = message.replace(str(eval("trigger." + triggeritem)).upper(), str("trigger" + triggeritem).upper()) + messagelist = spicemanip.main(message, "create") + for i in range(len(messagelist)): + if messagelist[i].upper() == str(eval("trigger." + botitem)).upper(): + messagelist[i] = str("trigger" + botitem).upper() + message = spicemanip.main(messagelist, 0) return message