This commit is contained in:
deathbybandaid 2019-06-01 13:55:55 -04:00
parent 659395b3eb
commit 6b305fdea0
2 changed files with 28 additions and 21 deletions

View File

@ -23,6 +23,7 @@ class SpiceBot_AI():
def __init__(self): def __init__(self):
self.load_commands = [0] self.load_commands = [0]
self.braindirs = []
self.dict = { self.dict = {
"counts": 0, "counts": 0,
"failcounts": 0, "failcounts": 0,
@ -57,9 +58,9 @@ class SpiceBot_AI():
def learn(self, braindirs): def learn(self, braindirs):
for braindir in braindirs: for braindir in braindirs:
if braindir not in self.braindirs:
self.braindirs.append(braindir)
dirnumber = max(self.load_commands) + 1 dirnumber = max(self.load_commands) + 1
# for parsefile in os.listdir(braindir):
# self.check_file_parse(parsefile)
tempbrain = tempfile.mkstemp()[1] tempbrain = tempfile.mkstemp()[1]
with open(tempbrain, 'w') as fileo: with open(tempbrain, 'w') as fileo:
fileo.write( fileo.write(

View File

@ -5,6 +5,8 @@ This is the SpiceBot AI system.
""" """
import sopel import sopel
import os
import sopel_modules.SpiceBot as SpiceBot import sopel_modules.SpiceBot as SpiceBot
@ -12,6 +14,10 @@ import sopel_modules.SpiceBot as SpiceBot
@sopel.module.rule('.*') @sopel.module.rule('.*')
def bot_events_complete_ai(bot, trigger): def bot_events_complete_ai(bot, trigger):
for braindir in SpiceBot.botai.braindirs:
for parsefile in os.listdir(braindir):
SpiceBot.botai.check_file_parse(parsefile)
if SpiceBot.botai.dict["counts"]: if SpiceBot.botai.dict["counts"]:
SpiceBot.logs.log('SpiceBot_AI', 'Registered %d %s files,' % (SpiceBot.botai.dict["counts"], 'aiml')) SpiceBot.logs.log('SpiceBot_AI', 'Registered %d %s files,' % (SpiceBot.botai.dict["counts"], 'aiml'))
SpiceBot.logs.log('SpiceBot_AI', '%d %s files failed to load' % (SpiceBot.botai.dict["failcounts"], 'aiml'), True) SpiceBot.logs.log('SpiceBot_AI', '%d %s files failed to load' % (SpiceBot.botai.dict["failcounts"], 'aiml'), True)