diff --git a/sopel_modules/SpiceBot/AI.py b/sopel_modules/SpiceBot/AI.py index 019828b..8c8afb2 100644 --- a/sopel_modules/SpiceBot/AI.py +++ b/sopel_modules/SpiceBot/AI.py @@ -23,7 +23,7 @@ class SpiceBot_AI(): def __init__(self): self.load_commands = [0] - self.braindirs = [] + self.braindirs = {} self.dict = { "counts": 0, "failcounts": 0, @@ -63,40 +63,21 @@ class SpiceBot_AI(): for braindir in braindirs: if braindir not in self.braindirs: self.braindirs.append(braindir) - dirnumber = max(self.load_commands) + 1 - self.load_commands.append(dirnumber) tempbrain = tempfile.mkstemp()[1] - # with open(tempbrain, 'w') as fileo: - # fileo.write( - # "" - # " \n" - # " \n" - # " LOAD AIML B\n" - # # " LOAD AIML {}\n" - # " \n" - # " \n" - # # "".format(str(dirnumber), os.path.join(braindir, "*")) - # "".format(os.path.join(braindir, "*.aiml")) - # ) - brainfile = open(tempbrain, 'a') - brainfile.write("") - brainfile.write(" \n") - brainfile.write(" \n") - brainfile.write(" LOAD AIML {}\n".format(str(dirnumber))) - brainfile.write(" \n") - brainfile.write(" \n") - brainfile.write("") - brainfile.close() + with open(tempbrain, 'w') as fileo: + fileo.write( + "" + " \n" + " \n" + " LOAD AIML B\n" + " \n" + " \n" + "".format(os.path.join(braindir, "*.aiml")) + ) self.aiml_kernel.learn(tempbrain) - # self.aiml_kernel.respond("LOAD AIML B") - for number in self.load_commands: - if number != 0: - self.aiml_kernel.respond("LOAD AIML {}").format(str(dirnumber)) - os.remove(tempbrain) + self.aiml_kernel.respond("LOAD AIML B") def on_message(self, bot, trigger, message): nick = Identifier(trigger.nick)