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

test

Browse Source
This commit is contained in:
deathbybandaid 2019-06-04 08:19:08 -04:00
parent 1760e527d7
commit dbb2700eac
1 changed files with 34 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
sopel_modules/SpiceBot/AI.py
View File

@ -22,8 +22,8 @@ class SpiceBot_AI_MainSection(StaticSection):
class SpiceBot_AI(): class SpiceBot_AI():
def __init__(self): def __init__(self):
self.load_commands = [0] self.load_commands = {0: None}
self.braindirs = {} self.braindirs = []
self.dict = { self.dict = {
"counts": 0, "counts": 0,
"failcounts": 0, "failcounts": 0,
@ -63,21 +63,40 @@ class SpiceBot_AI():
for braindir in braindirs: for braindir in braindirs:
if braindir not in self.braindirs: if braindir not in self.braindirs:
self.braindirs.append(braindir) self.braindirs.append(braindir)
dirnumber = max(self.load_commands.keys()) + 1
self.load_commands[dirnumber] = braindir
tempbrain = tempfile.mkstemp()[1] tempbrain = tempfile.mkstemp()[1]
with open(tempbrain, 'w') as fileo: # 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"
" <category>\n" # " <category>\n"
" <pattern>LOAD AIML B</pattern>\n" # " <pattern>LOAD AIML B</pattern>\n"
" <template>\n" # # " <pattern>LOAD AIML {}</pattern>\n"
" <learn>{}</learn>\n" # " <template>\n"
" </template>\n" # " <learn>{}</learn>\n"
" </category>\n" # " </template>\n"
"</aiml>".format(os.path.join(braindir, "*.aiml")) # " </category>\n"
) # # "</aiml>".format(str(dirnumber), os.path.join(braindir, "*"))
# "</aiml>".format(os.path.join(braindir, "*.aiml"))
# )
brainfile = open(tempbrain, 'a')
brainfile.write("<aiml version='1.0.1' encoding='UTF-8'>")
brainfile.write(" <!-- std-startup.xml -->\n")
brainfile.write(" <category>\n")
brainfile.write(" <pattern>LOAD AIML {}</pattern>\n".format(str(dirnumber)))
brainfile.write(" <template>\n")
brainfile.write(" <learn>{}</learn>\n".format(os.path.join(braindir, "*.aiml")))
brainfile.write(" </template>\n")
brainfile.write(" </category>\n")
brainfile.write("</aiml>")
brainfile.close()
self.aiml_kernel.learn(tempbrain) self.aiml_kernel.learn(tempbrain)
self.aiml_kernel.respond("LOAD AIML B") # self.aiml_kernel.respond("LOAD AIML B")
for number in self.load_commands.keys():
if number != 0:
self.aiml_kernel.respond("LOAD AIML {}").format(str(dirnumber))
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)