bitbot-3.11-fork/modules/karma.py

import re, time
import EventManager, Utils

REGEX_KARMA = re.compile("^(.*[^-+])[-+]*(\+{2,}|\-{2,})$")
KARMA_DELAY_SECONDS = 3

class Module(object):
    def __init__(self, bot, events, exports):
        self.bot = bot
        self.events = events
        events.on("new.user").hook(self.new_user)
        events.on("received.message.channel").hook(
            self.channel_message, priority=EventManager.PRIORITY_MONITOR)
        events.on("received.command.karma").hook(
            self.karma, help="Get your or someone else's karma",
            usage="[target]")
        events.on("received.command.resetkarma").hook(
            self.reset_karma, permission="resetkarma",
            min_args=1, help="Reset a specified karma to 0",
            usage="<target>")

        exports.add("channelset", {"setting": "karma-verbose",
            "help": "Disable/Enable automatically responding to "
            "karma changes", "validate": Utils.bool_or_none})

    def new_user(self, event):
        event["user"].last_karma = None

    def channel_message(self, event):
        match = re.match(REGEX_KARMA, event["message"].strip())
        if match and not event["action"]:
            verbose = event["channel"].get_setting("karma-verbose", False)
            if not event["user"].last_karma or (time.time()-event["user"
                    ].last_karma) >= KARMA_DELAY_SECONDS:
                target = match.group(1).lower().strip()
                if not target == event["user"].name and target:
                    positive = match.group(2)[0] == "+"
                    setting = "karma-%s" % target
                    karma = event["server"].get_setting(setting, 0)
                    if positive:
                        karma += 1
                    else:
                        karma -= 1
                    if karma:
                        event["server"].set_setting(setting, karma)
                    else:
                        event["server"].del_setting(setting)
                    if verbose:
                        self.events.on("send.stdout").call(
                            module_name="Karma", target=event["channel"],
                            message="%s now has %d karma" % (target, karma))
                    event["user"].last_karma = time.time()
                elif verbose:
                    if target:
                        self.events.on("send.stderr").call(
                            module_name="Karma", target=event["channel"],
                            message="You cannot change your own karma")
            elif verbose:
                self.events.on("send.stderr").call(module_name="Karma",
                    target=event["channel"],
                    message="Try again in a couple of seconds")

    def karma(self, event):
        if event["args"]:
            target = event["args"]
        else:
            target = event["user"].nickname
        karma = event["server"].get_setting("karma-%s" % target, 0)
        event["stdout"].write("%s has %s karma" % (target, karma))

    def reset_karma(self, event):
        setting = "karma-%s" % event["args_split"][0]
        karma = event["server"].get_setting(setting, 0)
        if karma == 0:
            event["stderr"].write("%s already has 0 karma" % event[
                "args_split"][0])
        else:
            event["server"].del_setting(setting)
            event["stdout"].write("Reset karma for %s" % event[
                "args_split"][0])