55 lines
2.2 KiB
Python
55 lines
2.2 KiB
Python
|
|
||
|
|
|
||
|
|
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)
|
||
|
|
|
||
|
|
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, target, 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:
|
||
|
|
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["channel"].mode_or_above(event["user"].nickname,
|
||
|
|
"o"):
|
||
|
|
self._set(self.channel_settings, event["channel"], event)
|
||
|
|
else:
|
||
|
|
event["stderr"].write("You do not have the modes required")
|