deathbybandaid/SpiceBot-migrated
1
0
Fork 0
mirror of https://github.com/SpiceBot/SpiceBot.git synced 2026-11-13 06:59:53 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

config setup

Browse Source
This commit is contained in:
deathbybandaid 2019-05-31 22:23:17 -04:00
parent dfe1007609
commit aba9c3380f
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
sopel_modules/SpiceBot/AI.py
View File

@ -31,11 +31,17 @@ class SpiceBot_AI():
def load_brain(self): def load_brain(self):
import sopel_modules import sopel_modules
braindirs = []
for plugin_dir in set(sopel_modules.__path__): for plugin_dir in set(sopel_modules.__path__):
configsdir = os.path.join(plugin_dir, "SpiceBot_Configs") configsdir = os.path.join(plugin_dir, "SpiceBot_Configs")
aimldir = os.path.join(configsdir, "aiml") aimldir = os.path.join(configsdir, "aiml")
aimlstarter = tempfile.mkstemp()[1] braindirs.append(aimldir)
with open(aimlstarter, 'w') as fileo:
# TODO add extra config
for braindir in braindirs:
tempbrain = tempfile.mkstemp()[1]
with open(tempbrain, 'w') as fileo:
fileo.write( fileo.write(
"<aiml version='1.0.1' encoding='UTF-8'>" "<aiml version='1.0.1' encoding='UTF-8'>"
" <!-- std-startup.xml -->\n" " <!-- std-startup.xml -->\n"
@ -45,11 +51,11 @@ class SpiceBot_AI():
" <learn>{}</learn>\n" " <learn>{}</learn>\n"
" </template>\n" " </template>\n"
" </category>\n" " </category>\n"
"</aiml>".format(os.path.join(aimldir, "*")) "</aiml>".format(os.path.join(braindir, "*"))
) )
self.aiml_kernel.learn(aimlstarter) self.aiml_kernel.learn(tempbrain)
self.aiml_kernel.respond("LOAD AIML B") self.aiml_kernel.respond("LOAD AIML B")
os.remove(aimlstarter) os.remove(tempbrain)
def on_message(self, bot, trigger, message): def on_message(self, bot, trigger, message):
nick = Identifier(trigger.nick) nick = Identifier(trigger.nick)