bitbot-3.11-fork/modules/badwords.py

import time
from src import ModuleManager, utils

class Module(ModuleManager.BaseModule):
    @utils.hook("received.command.badwordslist", channel_only=True)
    def badwords_list(self, event):
        """
        :help: List the badwords in the current channel
        :require_mode: o
        """
        badwords = event["target"].get_setting("badwords", [])
        badwords = ("(%d) %s" % (i+1, badword["pattern"]) for i, badword in
            enumerate(badwords))
        event["stdout"].write("%s: %s" % (event["target"].name,
            ", ".join(badwords)))

    @utils.hook("received.command.badwordsadd", channel_only=True, min_args=2)
    def badwords_add(self, event):
        """
        :help: Add a badword to the badwords list for the current channel
        :usage: kick|ban|kickban <pattern>
        :require_mode: o
        """
        action = event["args_split"][0].lower()
        if not action in ["kick", "ban", "kickban"]:
            raise utils.EventError("Unknown action '%s'" % action)

        badwords = event["target"].get_setting("badwords", [])
        badwords.append({
            "pattern": " ".join(event["args_split"][1:]).lower(),
            "action": action,
            "added_by": event["user"].nickname,
            "added_at": time.time()})
        event["target"].set_setting("badwords", badwords)
        event["stdout"].write("%s: added to badwords" % event["user"].nickname)

    @utils.hook("received.command.badwordsdel", channel_only=True, min_args=1)
    def badwords_del(self, event):
        """
        :help: Remove a badwords from the current channel's badwords list
        :usage: <index>
        :require_mode: o
        """
        index = event["args_split"][0]
        if not index.isdigit() or int(index) == 0:
            raise utils.EventError("%s: index must be a positive number" %
                event["user"].nickname)

        index_int = int(event["args_split"][0])-1
        badwords = event["target"].get_setting("badwords", [])
        if index_int >= len(badwords):
           raise utils.EventError("%s: unknown badwords index %s" % (
                event["user"].nickname, index))

        badwords.pop(index_int)
        event["target"].set_setting("badwords", badwords)
        event["stdout"].write("%s: removed from badwords" %
            event["user"].nickname)

    @utils.hook("received.message.channel")
    def channel_message(self, event):
        if event["channel"].mode_or_above(event["user"], "v"):
            return

        badwords = event["channel"].get_setting("badwords", [])
        message_lower = event["message"].lower()
        for badword in badwords:
            if badword["pattern"] in message_lower:
                kick = False
                ban = False
                if badword["action"] == "kick":
                    kick = True
                elif badword["action"] == "kickban":
                    ban = True
                    kick = True
                elif badword["action"] == "ban":
                    ban = True

                if ban:
                    event["channel"].send_ban("*!%s@%s" % (
                        event["user"].username,
                        event["user"].realname))
                if kick:
                    event["channel"].send_kick(event["user"].nickname,
                        "You said a badword!")