aboutsummaryrefslogtreecommitdiff
path: root/modules/print_activity.py
blob: 747d732f3c75a3e871c9f700146352a8a3015ab5 (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
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
79
80
81
82
83
import datetime
import EventManager

class Module(object):
    def __init__(self, bot):
        self.bot = bot

        bot.events.on("received").on("message").on("channel").hook(
            self.channel_message,
            priority=EventManager.EventCallback.PRIORITY_HIGH)
        bot.events.on("self").on("message").on("channel").hook(
            self.self_channel_message)

        bot.events.on("received").on("join").hook(self.join)
        bot.events.on("self").on("join").hook(self.self_join)

        bot.events.on("received").on("part").hook(self.part)
        bot.events.on("self").on("part").hook(self.self_part)

        bot.events.on("received").on("nick").hook(self.on_nick)
        bot.events.on("self").on("nick").hook(self.on_nick)

        bot.events.on("received").on("quit").hook(self.on_quit)

        bot.events.on("received").on("kick").hook(self.kick)
        bot.events.on("self").on("kick").hook(self.self_kick)

    def print_line(self, event, line):
        timestamp = datetime.datetime.now().isoformat()
        target = str(event["server"])
        if "channel" in event:
            target += event["channel"].name
        print("[%s] %s | %s" % (timestamp, target, line))

    def _on_message(self, event, nickname):
        if not self.bot.args.verbose:
            if event["action"]:
                self.print_line(event, "* %s %s" % (nickname, event["message"]))
            else:
                self.print_line(event, "<%s> %s" % (nickname, event["message"]))
    def channel_message(self, event):
        self._on_message(event, event["user"].nickname)
    def self_channel_message(self, event):
        self._on_message(event, event["server"].nickname)

    def _on_join(self, event, nickname):
        if not self.bot.args.verbose:
            self.print_line(event, "%s joined %s" % (nickname,
                event["channel"].name))
    def join(self, event):
        self._on_join(event, event["user"].nickname)
    def self_join(self, event):
        self._on_join(event, event["server"].nickname)

    def _on_part(self, event, nickname):
        if not self.bot.args.verbose:
            self.print_line(event, "%s left %s%s" % (nickname,
                event["channel"].name, "" if not event[
                "reason"] else ": %s" % event["reason"]))
    def part(self, event):
        self._on_part(event, event["user"].nickname)
    def self_part(self, event):
        self._on_part(event, event["server"].nickname)

    def on_nick(self, event):
        if not self.bot.args.verbose:
            self.print_line(event, "%s changed nickname to %s" % (
                event["old_nickname"], event["new_nickname"]))

    def on_quit(self, event):
        if not self.bot.args.verbose:
            self.print_line(event, "%s quit" % event["user"].nickname)

    def _on_kick(self, event, nickname):
        if not self.bot.args.verbose:
            self.print_line(event, "%s kicked %s from %s%s" % (
                event["user"].nickname, nickname, event["channel"].name,
                " (%s)" % event["reason"] if event[
                "reason"] else ""))
    def kick(self, event):
        self._on_kick(event, event["target_user"].nickname)
    def self_kick(self, event):
        self._on_kick(event, event["server"].nickname)