Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Also track BitBot's words in words.py

Browse source
This commit is contained in:
jesopo 2018-08-08 10:33:33 +01:00
parent e6321abd55
commit 2e9ce67586
1 changed files with 12 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
modules/words.py
View file

@ -5,6 +5,8 @@ class Module(object):
self.bot = bot self.bot = bot
bot.events.on("received").on("message").on("channel" bot.events.on("received").on("message").on("channel"
).hook(self.channel_message) ).hook(self.channel_message)
bot.events.on("self").on("message").on("channel"
).hook(self.self_channel_message)
bot.events.on("received").on("command").on("words" bot.events.on("received").on("command").on("words"
).hook(self.words, channel_only=True, ).hook(self.words, channel_only=True,
usage="<nickname>", help= usage="<nickname>", help=
@ -18,14 +20,14 @@ class Module(object):
help="Show who has used a tracked word the most", help="Show who has used a tracked word the most",
usage="<word>") usage="<word>")
def channel_message(self, event): def _channel_message(self, user, event):
words = list(filter(None, event["message_split"])) words = list(filter(None, event["message_split"]))
word_count = len(words) word_count = len(words)
user_words = event["channel"].get_user_setting( user_words = event["channel"].get_user_setting(
event["user"].nickname, "words", 0) user.nickname, "words", 0)
user_words += word_count user_words += word_count
event["channel"].set_user_setting(event["user"].nickname, event["channel"].set_user_setting(user.nickname,
"words", user_words) "words", user_words)
tracked_words = set(event["server"].get_setting( tracked_words = set(event["server"].get_setting(
@ -33,9 +35,14 @@ class Module(object):
for word in words: for word in words:
if word.lower() in tracked_words: if word.lower() in tracked_words:
setting = "word-%s" % word setting = "word-%s" % word
word_count = event["user"].get_setting(setting, 0) word_count = user.get_setting(setting, 0)
word_count += 1 word_count += 1
event["user"].set_setting(setting, word_count) user.set_setting(setting, word_count)
def channel_message(self, event):
self._channel_message(event["user"], event)
def self_channel_message(self, event):
self._channel_message(event["server"].get_user(
event["server"].nickname), event)
def words(self, event): def words(self, event):
if event["args_split"]: if event["args_split"]: