diff --git a/sopel_modules/SpiceBot/AI.py b/sopel_modules/SpiceBot/AI.py index 2bb5a65..d0ba3e7 100644 --- a/sopel_modules/SpiceBot/AI.py +++ b/sopel_modules/SpiceBot/AI.py @@ -31,11 +31,17 @@ class SpiceBot_AI(): def load_brain(self): import sopel_modules + braindirs = [] for plugin_dir in set(sopel_modules.__path__): configsdir = os.path.join(plugin_dir, "SpiceBot_Configs") aimldir = os.path.join(configsdir, "aiml") - aimlstarter = tempfile.mkstemp()[1] - with open(aimlstarter, 'w') as fileo: + braindirs.append(aimldir) + + # TODO add extra config + + for braindir in braindirs: + tempbrain = tempfile.mkstemp()[1] + with open(tempbrain, 'w') as fileo: fileo.write( "" " \n" @@ -45,11 +51,11 @@ class SpiceBot_AI(): " {}\n" " \n" " \n" - "".format(os.path.join(aimldir, "*")) + "".format(os.path.join(braindir, "*")) ) - self.aiml_kernel.learn(aimlstarter) + self.aiml_kernel.learn(tempbrain) self.aiml_kernel.respond("LOAD AIML B") - os.remove(aimlstarter) + os.remove(tempbrain) def on_message(self, bot, trigger, message): nick = Identifier(trigger.nick)