aboutsummaryrefslogtreecommitdiff
path: root/modules/words.py
blob: c0c8169b06f634d000eb0818a321e0c1bde342d4 (about) (plain) (blame)
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
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))