bitbot-3.11-fork/modules/print_activity.py

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.PRIORITY_HIGH)
        bot.events.on("self").on("message").on("channel").hook(
            self.self_channel_message)

        bot.events.on("received").on("notice").on("channel").hook(
            self.channel_notice, priority=EventManager.PRIORITY_HIGH)
        bot.events.on("received").on("notice").on("private").hook(
            self.private_notice, priority=EventManager.PRIORITY_HIGH)
        bot.events.on("received").on("server-notice").hook(
            self.server_notice, priority=EventManager.PRIORITY_HIGH)

        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)

        bot.events.on("received").on("topic").hook(self.on_topic)
        bot.events.on("received").on("numeric").on("333").hook(self.on_333)

    def print_line(self, event, line, channel=None):
        timestamp = datetime.datetime.now().isoformat()
        target = str(event["server"])
        if not channel == None:
            target += channel
        self.bot.log.info("%s | %s", [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"]),
                                channel=event["channel"].name)
            else:
                self.print_line(event, "<%s> %s" % (
                    nickname, event["message"]),
                                channel=event["channel"].name)

    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_notice(self, event, target):
        self.print_line(event, "(notice->%s) <%s> %s" % (
            target, event["user"].nickname, event["message"]))

    def channel_notice(self, event):
        self._on_notice(event, event["channel"].name)

    def private_notice(self, event):
        self._on_notice(event, event["server"].nickname)

    def server_notice(self, event):
        self.print_line(event, "(server notice) %s" % event["message"])

    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%s" % (event["user"].nickname,
                                                  "" if not event[
                                                      "reason"] else " (%s)" %
                                                                     event[
                                                                         "reason"]))

    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,
                "" if not event["reason"] else " (%s)" % event["reason"]))

    def kick(self, event):
        self._on_kick(event, event["target_user"].nickname)

    def self_kick(self, event):
        self._on_kick(event, event["server"].nickname)

    def _on_topic(self, event, setter, action, topic, channel):
        self.print_line(event, "topic %s by %s: %s" % (action, setter,
                                                       topic),
                        channel=channel.name)

    def on_topic(self, event):
        self._on_topic(event, event["user"].nickname, "changed",
                       event["topic"], event["channel"])

    def on_333(self, event):
        self._on_topic(event, event["setter"], "set",
                       event["channel"].topic, event["channel"])