From 51a52e2b0e54031cce5876f54d1d48c268b5441c Mon Sep 17 00:00:00 2001 From: jesopo Date: Wed, 26 Sep 2018 18:27:17 +0100 Subject: Switch to using @Utils.hook and docstrings for event hooks --- modules/set.py | 57 +++++++++++++++++++++++++++++++-------------------------- 1 file changed, 31 insertions(+), 26 deletions(-) (limited to 'modules/set.py') diff --git a/modules/set.py b/modules/set.py index 86cbe6bf..78001209 100644 --- a/modules/set.py +++ b/modules/set.py @@ -1,34 +1,9 @@ - +from src import Utils class Module(object): def __init__(self, bot, events, exports): - self.bot = bot self.exports = exports - events.on("received.command.set").hook(self.set, - usage=" ", help="Set a specified user setting") - events.on("received.command.get").hook(self.get, min_args=1, - usage="", help="Get a specified user setting") - - events.on("received.command.channelset").hook(self.channel_set, - channel_only=True, usage=" ", require_mode="o", - help="Set a specified setting for the current channel") - events.on("received.command.channelsetoverride").hook( - self.channel_set, channel_only=True, usage=" ", - permission="channelsetoverride", - help="Set a specified setting for the current channel") - events.on("received.command.channelget").hook(self.channel_get, - channel_only=True, usage="", min_args=1, - require_mode="o", help="Get a specified setting for the current " - "channel") - - events.on("received.command.serverset").hook(self.server_set, - usage=" ", permission="serverset", - help="Set a specified setting for the current server") - events.on("received.command.serverget").hook(self.server_get, - usage="", min_args=1, permission="serverget", - help="Get a specified setting for the current server") - def _set(self, settings, event, target): settings_dict = dict([(setting["setting"], setting ) for setting in settings]) @@ -50,13 +25,29 @@ class Module(object): else: event["stdout"].write("Available settings: %s" % ( ", ".join(settings_dict.keys()))) + @Utils.hook("received.command.set", usage=" ") def set(self, event): + """ + Set a specified user setting + """ self._set(self.exports.get_all("set"), event, event["user"]) + @Utils.hook("received.command.channelset", channel_only=True, + usage=" ", require_mode="o") + @Utils.hook("received.command.channelsetoverride", channel_only=True, + usage=" ", permission="channelsetoverride") def channel_set(self, event): + """ + Get a specified channel setting for the current channel + """ self._set(self.exports.get_all("channelset"), event, event["target"]) + @Utils.hook("received.command.serverset", usage=" ", + permission="serverset") def server_set(self, event): + """ + Set a specified server setting for the current server + """ self._set(self.exports.get_all("serverset"), event, event["server"]) def _get(self, event, setting, qualifier, value): @@ -66,17 +57,31 @@ class Module(object): else: event["stdout"].write("'%s' has no value set" % setting) + @Utils.hook("received.command.get", min_args=1, usage="") def get(self, event): + """ + Get a specified user setting + """ setting = event["args_split"][0] self._get(event, setting, "", event["user"].get_setting( setting, None)) + @Utils.hook("received.command.channelget", channel_only=True, + usage="", min_args=1, require_mode="o") def channel_get(self, event): + """ + Get a specified channel setting for the current channel + """ setting = event["args_split"][0] self._get(event, setting, " for %s" % event["target"].name, event["target"].get_setting(setting, None)) + @Utils.hook("received.command.serverget", usage="", min_args=1, + permission="serverget") def server_get(self, event): + """ + Get a specified server setting for the current server + """ setting = event["args_split"][0] self._get(event, setting, "", event["server"].get_setting( setting, None)) -- cgit v1.3.1-10-gc9f91