bitbot-3.11-fork/modules/set.py

class Module(object):
    def __init__(self, bot):
        self.bot = bot
        self.settings = {}
        self.channel_settings = {}
        bot.events.on("postboot").on("configure").on("set").hook(
            self.postboot_set)
        bot.events.on("postboot").on("configure").on("channelset"
            ).hook(self.postboot_channelset)
        bot.events.on("received").on("command").on("set").hook(
            self.set, help="Set a specified user setting")
        bot.events.on("received").on("command").on("channelset"
            ).hook(self.channel_set, channel_only=True,
            help="Set a specified setting for the current channel")

    def _postboot_set(self, settings, event):
        settings[event["setting"]] = {}
        settings[event["setting"]]["validate"] = event.get(
            "validate", lambda s: s)
        settings[event["setting"]]["help"] = event.get("help",
            "")
    def postboot_set(self, event):
        self._postboot_set(self.settings, event)
    def postboot_channelset(self, event):
        self._postboot_set(self.channel_settings, event)

    def _set(self, settings, event):
        if len(event["args_split"]) > 1:
            setting = event["args_split"][0].lower()
            if setting in settings:
                value = " ".join(event["args_split"][1:])
                value = settings[setting]["validate"](value)
                if not value == None:
                    event["target"].set_setting(setting, value)
                    event["stdout"].write("Saved setting")
                else:
                    event["stderr"].write("Invalid value")
            else:
                event["stderr"].write("Unknown setting")
        elif len(event["args_split"]) == 1:
            event["stderr"].write("Please provide a value")
        else:
            event["stdout"].write("Available settings: %s" % (
                ", ".join(settings.keys())))
    def set(self, event):
        self._set(self.settings, event["user"], event)

    def channel_set(self, event):
        if event["target"].mode_or_above(event["user"].nickname,
                "o"):
            self._set(self.channel_settings, event)
        else:
            event["stderr"].write("You do not have the modes required")