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

wordiest-prevent-highlight -> words-, apply to !wordusers too

Browse source
This commit is contained in:
jesopo 2020-01-23 14:15:26 +00:00
parent f6e6dd0339
commit 32f047c10e
1 changed files with 10 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
modules/words.py
View file

@ -14,6 +14,8 @@ SETTING = utils.BoolSetting("word-tracking",
@utils.export("set", SETTING) @utils.export("set", SETTING)
@utils.export("channelset", SETTING) @utils.export("channelset", SETTING)
@utils.export("channelset", utils.BoolSetting("words-prevent-highlight",
"Whether or not to prevent highlights in wordiest lists"))
class Module(ModuleManager.BaseModule): class Module(ModuleManager.BaseModule):
def _channel_message(self, user, event): def _channel_message(self, user, event):
if not event["channel"].get_setting("word-tracking", True if not event["channel"].get_setting("word-tracking", True
@ -109,6 +111,12 @@ class Module(ModuleManager.BaseModule):
event["stdout"].write("Tracked words: %s" % ", ".join( event["stdout"].write("Tracked words: %s" % ", ".join(
event["server"].get_setting("tracked-words", []))) event["server"].get_setting("tracked-words", [])))
def _get_nickname(self, server, target, nickname):
nickname = server.get_user(nickname).nickname
if target.get_setting("words-prevent-highlight", True):
nickname = utils.prevent_highlight(nickname)
return nickname
@utils.hook("received.command.wordusers", min_args=1) @utils.hook("received.command.wordusers", min_args=1)
def word_users(self, event): def word_users(self, event):
""" """
@ -122,19 +130,13 @@ class Module(ModuleManager.BaseModule):
items = [(word_user[0], word_user[1]) for word_user in word_users] items = [(word_user[0], word_user[1]) for word_user in word_users]
word_users = dict(items) word_users = dict(items)
top_10 = utils.top_10(word_users, top_10 = utils.top_10(word_users,
convert_key=lambda nickname: convert_key=lambda nickname: self._get_nickname(
event["server"].get_user(nickname).nickname) event["server"], event["target"], nickname))
event["stdout"].write("Top '%s' users: %s" % (word, event["stdout"].write("Top '%s' users: %s" % (word,
", ".join(top_10))) ", ".join(top_10)))
else: else:
event["stderr"].write("That word is not being tracked") event["stderr"].write("That word is not being tracked")
def _get_nickname(self, server, target, nickname):
nickname = server.get_user(nickname).nickname
if target.get_setting("wordiest-prevent-highlight", True):
nickname = utils.prevent_highlight(nickname)
return nickname
@utils.hook("received.command.wordiest") @utils.hook("received.command.wordiest")
def wordiest(self, event): def wordiest(self, event):
""" """