1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
class Module(object):
def __init__(self, bot):
self.bot = bot
bot.events.on("new").on("server").hook(self.new_server)
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,
usage="<nickname>")
bot.events.on("received").on("command").on("trackword"
).hook(self.track_word, min_args=1,
help="Start tracking a word", usage="<word>",
permission="track-word")
bot.events.on("received").on("command").on("wordusers"
).hook(self.word_users, min_args=1,
help="Show who has used a tracked word the most",
usage="<word>")
def new_server(self, event):
event["server"].tracked_words = set([])
settings = event["server"].find_settings("word-%")
for word, _ in settings:
word = word.split("word-", 1)[1]
event["server"].tracked_words.add(word)
def channel_message(self, event):
words = list(filter(None, event["message_split"]))
word_count = len(words)
user_words = event["user"].get_setting("words", {})
if not event["channel"].name in user_words:
user_words[event["channel"].name] = 0
user_words[event["channel"].name] += word_count
event["user"].set_setting("words", user_words)
for word in words:
if word.lower() in event["server"].tracked_words:
setting = "word-%s" % word
tracked_word = event["server"].get_setting(setting, {})
if not event["user"].name in tracked_word:
tracked_word[event["user"].name] = 0
tracked_word[event["user"].name] += 1
event["server"].set_setting(setting, tracked_word)
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))
def track_word(self, event):
word = event["args_split"][0].lower()
if not word in event["server"].tracked_words:
event["server"].tracked_words.add(word)
event["server"].set_setting("word-%s" % word, {})
event["stdout"].write("Now tracking '%s'" % word)
else:
event["stderr"].wrote("Already tracking '%s'" % word)
def word_users(self, event):
word = event["args_split"][0].lower()
if word in event["server"].tracked_words:
word_users = event["server"].get_setting("word-%s" % word)
top_10 = sorted(word_users.keys())
top_10 = sorted(top_10, key=word_users.get, reverse=True)[:10]
top_10 = ", ".join("%s (%d)" % (nickname, word_users[nickname
]) for nickname in top_10)
event["stdout"].write("Top '%s' users: %s" % (word, top_10))
else:
event["stderr"].write("That word is not being tracked")
|
