adammcclure
/
babili-bot
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fixed user name bug, made saving memories threading, added oofscoring, added logging

This commit is contained in:
aewens 2018-10-01 12:10:25 -04:00
parent 4e0978b534
commit b56de77a3f
7 changed files with 135 additions and 63 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.gitignore vendored
View File

@ -2,7 +2,7 @@
settings.json
settings.demo.json
data/*.json
logs/*.log
logs/*.log*
irc/
# ---> Python

30
actions/__init__.py
View File

@ -1,8 +1,8 @@
from actions.botlist import botlist
from actions.web import summon, whois
from actions.access import banish, pardon
from actions.control import puppet, inject, nomad
from actions.stupid import hmm, hmmscore, hmmscoreboard
from actions.web import summon, whois, how_dare_you
from actions.stupid import score_word, wordscore, wordscoreboard
actions = [
{
@ -20,6 +20,11 @@ actions = [
"pattern": "/!summon \S+ .+/",
"callback": summon
},
{
"type": "response",
"pattern": "/!summon \S+$/",
"callback": how_dare_you
},
{
"type": "response",
"pattern": "/!banish \S+ .+/",
@ -48,17 +53,32 @@ actions = [
{
"type": "response",
"pattern": "/hm+/",
"callback": hmm
"callback": score_word("hmm", "hm+")
},
{
"type": "response",
"pattern": "/!hmmscore(\s|$)/",
"callback": hmmscore
"callback": wordscore("hmm")
},
{
"type": "response",
"pattern": "!hmmscoreboard",
"callback": hmmscoreboard
"callback": wordscoreboard("hmm")
},
{
"type": "response",
"pattern": "/o+f/",
"callback": score_word("oof", "o+f")
},
{
"type": "response",
"pattern": "/!oofscore(\s|$)/",
"callback": wordscore("oof")
},
{
"type": "response",
"pattern": "!oofscoreboard",
"callback": wordscoreboard("oof")
},
{
"type": "response",

4
actions/access.py
View File

@ -16,7 +16,7 @@ def banish(self, name, source, response):
"when": datetime.now().timestamp()
}
self.bot.save_memories()
self.bot.thread(self.bot.save_memories)
confirmation = "{} has been banished for reason: {}".format(user, reason)
self.bot.send_message(source, confirmation)
@ -31,7 +31,7 @@ def pardon(self, name, source, response):
del self.bot.memories["users"][user]["blacklist"]
self.bot.save_memories()
self.bot.thread(self.bot.save_memories)
confirmation = "{} has been pardoned".format(user)
self.bot.send_message(source, confirmation)

51
actions/stupid.py
View File

@ -1,11 +1,15 @@
import re
import operator
def hmm(self, name, source, response):
def capitalize(word):
return word[0].upper() + word[1:]
def score_word(word, regex):
def wording(self, name, source, response):
check = response.lower().strip()
botnick = self.bot.botnick
pattern = re.compile("hm+")
pattern = re.compile(regex)#"hm+")
matches = re.findall(pattern, check)
maximum = 10
score = len(matches) if len(matches) <= maximum else maximum
@ -16,18 +20,22 @@ def hmm(self, name, source, response):
if name not in self.bot.memories["users"]:
self.bot.memories["users"][name] = dict()
if "hmmscore" not in self.bot.memories["users"][name]:
self.bot.memories["users"][name]["hmmscore"] = 0
keyword = "{}score".format(word)
current_score = self.bot.memories["users"][name]["hmmscore"]
self.bot.memories["users"][name]["hmmscore"] = current_score + score
if keyword not in self.bot.memories["users"][name]:
self.bot.memories["users"][name][keyword] = 0
self.bot.save_memories()
current_score = self.bot.memories["users"][name][keyword]
self.bot.memories["users"][name][keyword] = current_score + score
def hmmscore(self, name, source, response):
self.bot.thread(self.bot.save_memories)
return wording
def wordscore(word):
def scoring(self, name, source, response):
botnick = self.bot.botnick
score = 0
score_format = "Hmm score for '{}': {}"
score_format = "%s score for '{}': {}" % (capitalize(word))
if " " in response:
name = response.split(" ", 1)[1].strip()
@ -36,29 +44,38 @@ def hmmscore(self, name, source, response):
self.bot.send_message(source, score_format.format(name, score))
return
if "hmmscore" in self.bot.memories["users"][name]:
score = self.bot.memories["users"][name]["hmmscore"]
keyword = "{}score".format(word)
if keyword not in self.bot.memories["users"][name]:
self.bot.send_message(source, score_format.format(name, score))
return
def hmmscoreboard(self, name, source, response):
score = self.bot.memories["users"][name][keyword]
self.bot.send_message(source, score_format.format(name, score))
return scoring
def wordscoreboard(word):
def scoreboard(self, name, source, response):
botnick = self.bot.botnick
hmmscores = list()
scores = list()
for user, values in self.bot.memories["users"].items():
hmmscores.append({
scores.append({
"name": user,
"score": values.get("hmmscore", 0)
"score": values.get("{}score".format(word), 0)
})
size = 3
start = -size
sort_scores = sorted(hmmscores, key=lambda k: k["score"])
sort_scores = sorted(scores, key=lambda k: k["score"])
top_scores = sort_scores[start:][::-1]
leaders = " | ".join([
"{} {}".format(ts["name"], ts["score"]) for ts in top_scores
])
self.bot.send_message(source, "Hmm Score Leaderboard: {}".format(leaders))
response = "{} Score Leaderboard: {}".format(capitalize(word), leaders)
self.bot.send_message(source, response)
return scoreboard

5
actions/web.py
View File

@ -74,6 +74,11 @@ def summon(self, name, source, response):
confirmation = "{}: You have summoned {}".format(name, user)
self.bot.send_message(source, confirmation)
def how_dare_you(self, name, source, response):
user = response.split("!summon ")[1]
rude = "{}: You think you can just summon someone without a reason? Rude."
self.bot.send_message(source, rude.format(user))
def whois(self, name, source, response):
botnick = self.bot.botnick
domain = response.split("!whois ")[1]

10
app.py
View File

@ -68,17 +68,26 @@ def handle_mode(channel, mode):
try_to_king_me(channel)
def handle_invite(channel, name):
changed = False
if channel in kingme:
try_to_king_me(channel)
users = bot.memories["users"]
if name not in users:
bot.memories["users"][name] = dict()
changed = True
if "invites" not in users[name]:
bot.memories["users"][name]["invites"] = list()
changed = True
if channel not in bot.memories["users"][name]["invites"]:
bot.memories["users"][name]["invites"].append(channel)
changed = True
if changed:
bot.thread(bot.save_memories)
def handle_kick(name):
users = bot.memories["users"]
@ -86,6 +95,7 @@ def handle_kick(name):
bot.memories["users"][name] = dict()
bot.memories["users"][name]["kicker"] = True
bot.thread(bot.save_memories)
def handle_message(name, source, response):
responses.parse(name, source, response)

40
bot/core.py
View File

@ -3,6 +3,7 @@ import json
import socket
import os.path
import logging
from threading import Thread
from logging.handlers import TimedRotatingFileHandler
logging.basicConfig(
@ -76,12 +77,11 @@ class Bot:
return
raw_users = user_listing[1].split(" \r\n")[0].split(" ")
prefix_filter = lambda u: u[1:] if "~" in u or "@" in u else u
users = list(filter(prefix_filter, raw_users))
# prefix_filter = lambda u: u[1:] if "~" in u or "@" in u or "+" in u else u
users = list(filter(self.parse_name, raw_users))
remember = self.memories["users"]
for user in users:
if user[0] == "~" or user[0] == "@":
user = user[1:]
user = self.parse_name(user)
if user not in remember:
self.memories["users"][user] = dict()
@ -98,21 +98,31 @@ class Bot:
def get_name(self, text):
return text.split("!", 1)[0][1:]
def parse_name(self, name):
if name[0] == "~" or name[0] == "@" or name[0] == "+":
return name[1:]
else:
return name
def parse(self, message):
before, after = message.split("PRIVMSG ", 1)
name = self.get_name(before)
name = self.parse_name(self.get_name(before))
source, response = after.split(" :", 1)
return name, source, response
def handle_mode(self, message):
before, after = message.split("MODE ", 1)
name = self.get_name(before)
name = self.parse_name(self.get_name(before))
channel, mode = after.split(" ")[:2]
return channel, mode
def handle_rename(self, message):
before, new_name = message.split("NICK ", 1)
name = self.get_name(before)
new_name = self.parse_name(new_name)
name = self.parse_name(name)
user = self.memories["users"][name]
del self.memories["users"][name]
self.memories["users"][new_name] = user
@ -120,19 +130,19 @@ class Bot:
def handle_invite(self, message):
before, after = message.split("INVITE ", 1)
name = self.get_name(before)
name = self.parse_name(self.get_name(before))
channel = after.split(":", 1)[1]
self.join(channel)
return channel, name
def handle_kick(self, message):
before, after = message.split("KICK ", 1)
name = self.get_name(before)
name = self.parse_name(self.get_name(before))
return name
def handle_join(self, message):
before, after = message.split("JOIN ", 1)
user = self.get_name(before)
user = self.parse_name(self.get_name(before))
if user not in self.memories["users"]:
self.memories["users"][user] = dict()
@ -141,7 +151,7 @@ class Bot:
def handle_part(self, message):
before, after = message.split("PART ", 1)
user = self.get_name(before)
user = self.parse_name(self.get_name(before))
return user
def load_memories(self, location):
@ -156,6 +166,11 @@ class Bot:
with open(path, "r") as f:
self.memories = json.loads(f.read())
def thread(self, fn, *args):
print((self, *args))
t = Thread(target=fn, args=args)
t.start()
def save_memories(self):
with open(self.memories_path, "w") as f:
try:
@ -243,6 +258,11 @@ class Bot:
message = self.ircsock.recv(self.recv_size).decode()
message = message.strip(self.splitter)
print(message)
if "new_nick' commands." in message:
self.logger.warning(message)
self.send_message(self.author, "ERROR: {}".format(message))
self.logger.debug(message)
if "raw" in callback: