diff --git a/sopel_modules/SpiceBot/AI.py b/sopel_modules/SpiceBot/AI.py index 72fbd60..8fbaec8 100644 --- a/sopel_modules/SpiceBot/AI.py +++ b/sopel_modules/SpiceBot/AI.py @@ -22,6 +22,7 @@ class SpiceBot_AI_MainSection(StaticSection): class SpiceBot_AI(): def __init__(self): + self.load_commands = [0] self.dict = { "counts": 0, "failcounts": 0, @@ -56,6 +57,7 @@ class SpiceBot_AI(): def learn(self, braindirs): for braindir in braindirs: + dirnumber = max(self.load_commands) + 1 for parsefile in os.listdir(braindir): self.check_file_parse(parsefile) tempbrain = tempfile.mkstemp()[1] @@ -64,15 +66,17 @@ class SpiceBot_AI(): "" " \n" " \n" - " LOAD AIML B\n" + " LOAD AIML {}\n" " \n" " \n" - "".format(os.path.join(braindir, "*")) + "".format(str(dirnumber), os.path.join(braindir, "*")) ) 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) def on_message(self, bot, trigger, message):