added words.py for keeping track of how many words people have said.

This commit is contained in:
jesopo 2016-04-05 18:43:17 +01:00
parent e1188e153a
commit 27c0de701c
No known key found for this signature in database
GPG key ID: 0BBDEB2AEFCFFCB3

32
modules/words.py Normal file
View file

@ -0,0 +1,32 @@
class Module(object):
def __init__(self, bot):
self.bot = bot
bot.events.on("received").on("message").on("channel"
).hook(self.channel_message)
bot.events.on("received").on("command").on("words"
).hook(self.words, channel_only=True)
def channel_message(self, event):
word_count = len(list(filter(None, event["message_split"
])))
words = event["user"].get_setting("words", {})
if not event["channel"].name in words:
words[event["channel"].name] = 0
words[event["channel"].name] += word_count
event["user"].set_setting("words", words)
def words(self, event):
if event["args_split"]:
target = event["server"].get_user(event["args_split"
][0])
else:
target = event["user"]
words = target.get_setting("words", {})
this_channel = words.get(event["target"].name, 0)
total = 0
for channel in words:
total += words[channel]
event["stdout"].write("%s has used %d words (%d in %s)" % (
target.nickname, total, this_channel, event["target"].name))