From 6920c7db1be42f8f1c848f0af361f109d47c2def Mon Sep 17 00:00:00 2001 From: deathbybandaid Date: Wed, 5 Jun 2019 15:42:33 -0400 Subject: [PATCH] counts --- sopel_modules/SpiceBot/AI.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/sopel_modules/SpiceBot/AI.py b/sopel_modules/SpiceBot/AI.py index debf743..2ac2cc2 100644 --- a/sopel_modules/SpiceBot/AI.py +++ b/sopel_modules/SpiceBot/AI.py @@ -26,8 +26,8 @@ class SpiceBot_AI(): self.setup_ai() self.braindirs = [] self.dict = { - "counts": 0, - "failcounts": 0, + "patterncount": 0, + "filecounts": 0, "sessioncache": {}, "files": {} } @@ -94,6 +94,16 @@ class SpiceBot_AI(): for braindir in braindirs: if braindir not in self.braindirs: self.braindirs.append(braindir) + + # Count matches + for pathname in os.listdir(braindir): + self.dict["filecounts"] += 1 + aimlfile = os.path.join(braindir, pathname) + data = open(aimlfile).read() + count = data.count('pattern') + count = count / 2 + self.dict["patterncount"] += count + tempbrain = tempfile.mkstemp()[1] with open(tempbrain, 'w') as fileo: fileo.write(