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

test parsing

Browse Source
This commit is contained in:
deathbybandaid 2019-06-04 13:54:43 -04:00
parent de21eb9367
commit c5a4c559cf
1 changed files with 19 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
sopel_modules/SpiceBot/AI.py
View File

@ -10,6 +10,7 @@ from sopel.config.types import StaticSection, ListAttribute
import os import os
import tempfile import tempfile
import aiml import aiml
import untangle
from .Database import db as botdb from .Database import db as botdb
from .Logs import logs from .Logs import logs
@ -78,6 +79,24 @@ class SpiceBot_AI():
self.aiml_kernel.learn(tempbrain) self.aiml_kernel.learn(tempbrain)
self.aiml_kernel.respond("LOAD AIML B") self.aiml_kernel.respond("LOAD AIML B")
def count_aiml_coms(self):
filepathlist = []
for braindir in self.braindirs:
for pathname in os.listdir(braindir):
path = os.path.join(braindir, pathname)
if os.path.isfile(path) and path.endswith('.aiml'):
filepathlist.append(str(path))
for aimlfile in filepathlist:
try:
aiml_dict = untangle.parse(aimlfile)
logs.log('SpiceBot_AI', aiml_dict)
self.dict["counts"] += 1
except Exception as e:
logs.log('SpiceBot_AI', "Error loading %s: %s (%s)" % ('aiml', e, aimlfile))
self.dict["failcounts"] += 1
def on_message(self, bot, trigger, message): def on_message(self, bot, trigger, message):
nick = Identifier(trigger.nick) nick = Identifier(trigger.nick)
nick_id = botdb.get_nick_id(nick, create=True) nick_id = botdb.get_nick_id(nick, create=True)
@ -128,13 +147,5 @@ class SpiceBot_AI():
sessionData = self.aiml_kernel.getSessionData(nick_id) sessionData = self.aiml_kernel.getSessionData(nick_id)
botdb.set_nick_value(nick, 'botai', sessionData) botdb.set_nick_value(nick, 'botai', sessionData)
def check_file_parse(self, parsefile):
try:
self.aiml_parser.parse(parsefile)
self.dict["counts"] += 1
except Exception as e:
logs.log('SpiceBot_AI', "Error loading %s: %s (%s)" % ('aiml', e, parsefile))
self.dict["failcounts"] += 1
botai = SpiceBot_AI() botai = SpiceBot_AI()