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

pronouns

Browse Source
This commit is contained in:
deathbybandaid 2019-06-05 11:58:24 -04:00
parent 198017d604
commit 1fde43d79c
1 changed files with 63 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

69
sopel_modules/SpiceBot/AI.py
View File

@ -73,15 +73,16 @@ class SpiceBot_AI():
self.save_brain() self.save_brain()
def load_bot_values(self): def load_bot_values(self):
# sopel nick
self.aiml_kernel.setBotPredicate("nick", botconfig.nick) self.aiml_kernel.setBotPredicate("nick", botconfig.nick)
# gender
self.aiml_kernel.setBotPredicate("gender", botconfig.SpiceBot_AI.gender.lower()) self.aiml_kernel.setBotPredicate("gender", botconfig.SpiceBot_AI.gender.lower())
if botconfig.SpiceBot_AI.gender.lower() == "male": if botconfig.SpiceBot_AI.gender.lower() not in ["male", "female"]:
pronoun = "he" self.aiml_kernel.setBotPredicate("gendertype", "item")
elif botconfig.SpiceBot_AI.gender.lower() == "female":
pronoun = "she"
else: else:
pronoun = "it" self.aiml_kernel.setBotPredicate("gendertype", botconfig.SpiceBot_AI.gender.lower())
self.aiml_kernel.setBotPredicate("pronoun", pronoun)
def learn(self, braindirs): def learn(self, braindirs):
for braindir in braindirs: for braindir in braindirs:
@ -119,23 +120,79 @@ class SpiceBot_AI():
return aiml_response return aiml_response
def bot_message_precipher(self, bot, trigger, message): def bot_message_precipher(self, bot, trigger, message):
# uppercase input
message = message.upper() message = message.upper()
# punctuation
puct_dict = {"!": "exclamationmark", ".": "period", "?": "questionmark", ",": "comma"} puct_dict = {"!": "exclamationmark", ".": "period", "?": "questionmark", ",": "comma"}
for puctuation in puct_dict.keys(): for puctuation in puct_dict.keys():
message = message.replace(puctuation, puct_dict[puctuation]) message = message.replace(puctuation, puct_dict[puctuation])
# bot items
for botitem in ["nick"]: for botitem in ["nick"]:
message = message.replace(str(eval("bot." + botitem)).upper(), str("bot" + botitem).upper()) message = message.replace(str(eval("bot." + botitem)).upper(), str("bot" + botitem).upper())
for triggeritem in ["nick", "sender"]: for triggeritem in ["nick", "sender"]:
message = message.replace(str(eval("trigger." + triggeritem)).upper(), str("trigger" + triggeritem).upper()) message = message.replace(str(eval("trigger." + triggeritem)).upper(), str("trigger" + triggeritem).upper())
return message return message
def bot_message_decipher(self, bot, trigger, aiml_response): def bot_message_decipher(self, bot, trigger, aiml_response):
# bot items
for botitem in ["nick"]: for botitem in ["nick"]:
aiml_response = aiml_response.replace("bot" + botitem, str(eval("bot." + botitem))) aiml_response = aiml_response.replace("bot" + botitem, str(eval("bot." + botitem)))
# trigger items
for triggeritem in ["nick", "sender"]: for triggeritem in ["nick", "sender"]:
aiml_response = aiml_response.replace("trigger" + triggeritem, str(eval("trigger." + triggeritem))) aiml_response = aiml_response.replace("trigger" + triggeritem, str(eval("trigger." + triggeritem)))
# pronouns
botgendertype = self.aiml_kernel.getBotPredicate("gendertype")
pronounsdict = {
"male": {
"main": "he",
"possess": "his",
"self": "himself",
},
"female": {
"main": "her",
"possess": "hers",
"self": "herself",
},
"item": {
"main": "it",
"possess": "its",
"self": "itself",
},
"point": {
"main": "you",
"possess": "yours",
"self": "yourself",
},
"group": {
"main": "them",
"possess": "theirs",
"self": "themselves",
},
}
for pronounitem in pronounsdict[botgendertype].keys():
aiml_response = aiml_response.replace("BOTPRONOUN" + pronounitem, pronounsdict[botgendertype][pronounitem])
triggergendertype = self.getPredicate("gender", trigger.nick)
if triggergendertype == "":
triggergendertype = "point"
for pronounitem in pronounsdict[triggergendertype].keys():
aiml_response = aiml_response.replace("TRIGGERPRONOUN" + pronounitem, pronounsdict[triggergendertype][pronounitem])
return aiml_response return aiml_response
def getPredicate(self, predicate, nick, nick_id=None):
if not nick_id:
nick = Identifier(nick)
nick_id = botdb.get_nick_id(nick, create=True)
self.aiml_kernel.getPredicate(predicate, nick_id)
def check_user_import(self, nick, nick_id=None): def check_user_import(self, nick, nick_id=None):
if not nick_id: if not nick_id:
nick = Identifier(nick) nick = Identifier(nick)