deathbybandaid/SpiceBot-migrated
1
0
Fork 0
mirror of https://github.com/SpiceBot/SpiceBot.git synced 2026-11-10 21:50:59 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

not today

Browse Source
This commit is contained in:
deathbybandaid 2019-06-04 15:56:19 -04:00
parent 8ad0e98e9e
commit 576a98749d
1 changed files with 0 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
sopel_modules/SpiceBot/AI.py
View File

@ -10,7 +10,6 @@ from sopel.config.types import StaticSection, ListAttribute
import os
import tempfile
import aiml
from bs4 import BeautifulSoup
from .Database import db as botdb
@ -40,7 +39,6 @@ class SpiceBot_AI():
# Learn responses
self.load_brain()
self.count_aiml_coms()
def load_brain(self):
import sopel_modules
@ -52,7 +50,6 @@ 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)
@ -81,46 +78,6 @@ class SpiceBot_AI():
self.aiml_kernel.learn(tempbrain)
self.aiml_kernel.respond("LOAD AIML B")
def count_aiml_coms(self):
self.dict["counts"] = 0
self.dict["failcounts"] = 0
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:
filereadgood = True
xmldoc = BeautifulSoup(aimlfile)
if aimlfile not in self.dict["files"]:
self.dict["files"][aimlfile] = xmldoc
return
aiml_file_lines = []
aiml_file = open(aimlfile, 'r')
lines = aiml_file.readlines()
for line in lines:
aiml_file_lines.append(str(line))
aiml_file.close()
detected_lines = []
for line in aiml_file_lines:
if line.startswith("<pattern>"):
line = line.replace("<pattern>", '')
line = line.replace("</pattern>", '')
detected_lines.append(line)
if not len(detected_lines):
filereadgood = False
if filereadgood:
self.dict["counts"] += 1
if aimlfile not in self.dict["files"]:
self.dict["files"][aimlfile] = detected_lines
else:
self.dict["failcounts"] += 1
def on_message(self, bot, trigger, message):
nick = Identifier(trigger.nick)
nick_id = botdb.get_nick_id(nick, create=True)