From c5a4c559cf84693bf79ee95395ff9c4a07f12ba5 Mon Sep 17 00:00:00 2001 From: deathbybandaid Date: Tue, 4 Jun 2019 13:54:43 -0400 Subject: [PATCH] test parsing --- sopel_modules/SpiceBot/AI.py | 27 +++++++++++++++++++-------- 1 file changed, 19 insertions(+), 8 deletions(-) diff --git a/sopel_modules/SpiceBot/AI.py b/sopel_modules/SpiceBot/AI.py index c88ef4b..ee1ceeb 100644 --- a/sopel_modules/SpiceBot/AI.py +++ b/sopel_modules/SpiceBot/AI.py @@ -10,6 +10,7 @@ from sopel.config.types import StaticSection, ListAttribute import os import tempfile import aiml +import untangle from .Database import db as botdb from .Logs import logs @@ -78,6 +79,24 @@ class SpiceBot_AI(): self.aiml_kernel.learn(tempbrain) 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): nick = Identifier(trigger.nick) nick_id = botdb.get_nick_id(nick, create=True) @@ -128,13 +147,5 @@ class SpiceBot_AI(): sessionData = self.aiml_kernel.getSessionData(nick_id) 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()