From 9829751b4a8a27da3bc1bda78dc9c38a24411129 Mon Sep 17 00:00:00 2001 From: deathbybandaid Date: Tue, 4 Jun 2019 14:00:51 -0400 Subject: [PATCH] counts --- sopel_modules/SpiceBot/AI.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/sopel_modules/SpiceBot/AI.py b/sopel_modules/SpiceBot/AI.py index ee1ceeb..b8c37d7 100644 --- a/sopel_modules/SpiceBot/AI.py +++ b/sopel_modules/SpiceBot/AI.py @@ -40,6 +40,7 @@ class SpiceBot_AI(): # Learn responses self.load_brain() + self.count_aiml_coms() def load_brain(self): import sopel_modules @@ -51,6 +52,7 @@ class SpiceBot_AI(): # learn directories self.learn(braindirs) + self.count_aiml_coms() def load_bot_values(self, bot): self.aiml_kernel.setBotPredicate("nick", bot.nick) @@ -88,13 +90,18 @@ class SpiceBot_AI(): filepathlist.append(str(path)) for aimlfile in filepathlist: + filereadgood = True try: - aiml_dict = untangle.parse(aimlfile) - logs.log('SpiceBot_AI', aiml_dict) - self.dict["counts"] += 1 + dict_from_file = untangle.parse(aimlfile) + logs.log('SpiceBot_AI', dict_from_file) except Exception as e: + filereadgood = False logs.log('SpiceBot_AI', "Error loading %s: %s (%s)" % ('aiml', e, aimlfile)) + + if filereadgood and isinstance(dict_from_file, dict): + self.dict["counts"] += 1 + else: self.dict["failcounts"] += 1 def on_message(self, bot, trigger, message):