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

verify settings get set

Browse Source
This commit is contained in:
deathbybandaid 2019-06-05 08:58:01 -04:00
parent 0ce1c8e0e9
commit be2d0eea2a
2 changed files with 16 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
sopel_modules/SpiceBot/AI.py
View File

@ -39,10 +39,19 @@ class SpiceBot_AI():
# Don't warn for no matches
self.aiml_kernel._verboseMode = False
# Config
self.botai_config()
# Learn responses
self.load_saved_brain()
self.load_brain()
# Load bot values
self.load_bot_values()
def botai_config(self):
botconfig.define_section("SpiceBot_AI", SpiceBot_AI_MainSection, validate=False)
def load_saved_brain(self):
if os.path.isfile(botconfig.config.aibrain):
self.aiml_kernel.bootstrap(brainFile=botconfig.config.aibrain)
@ -56,18 +65,17 @@ class SpiceBot_AI():
aimldir = os.path.join(configsdir, "aiml")
braindirs.append(aimldir)
if len(botconfig.SpiceBot_AI.extra):
for extrabraindir in botconfig.SpiceBot_AI.extra:
braindirs.append(extrabraindir)
# learn directories
self.learn(braindirs)
self.save_brain()
def load_bot_values(self, bot):
self.aiml_kernel.setBotPredicate("nick", bot.nick)
self.aiml_kernel.setBotPredicate("gender", bot.config.SpiceBot_AI.gender)
def load_extras(self, bot):
if len(bot.config.SpiceBot_AI.extra):
self.learn(self, bot.config.SpiceBot_AI.extra)
self.save_brain()
def load_bot_values(self):
self.aiml_kernel.setBotPredicate("nick", botconfig.nick)
self.aiml_kernel.setBotPredicate("gender", botconfig.SpiceBot_AI.gender)
def learn(self, braindirs):
for braindir in braindirs:

2
sopel_modules/SpiceBot/setup.py
View File

@ -81,5 +81,3 @@ def setup(bot):
logs.log('SpiceBot_AI', "Setting Up Configuration")
bot.config.define_section("SpiceBot_AI", SpiceBot_AI_MainSection, validate=False)
botai.load_extras(bot)
botai.load_bot_values(bot)