bitbot-3.11-fork/modules/set.py

142 lines
5.4 KiB
Python
Raw Normal View History

from src import ModuleManager, utils
2018-08-28 13:55:57 +00:00
CHANNELSET_HELP = "Get a specified channel setting for the current channel"
CHANNELSETADD_HELP = ("Add to a specified channel setting for the "
"current channel")
class Module(ModuleManager.BaseModule):
def _set(self, category, event, target, array):
settings = self.exports.get_all(category)
settings_dict = dict([(setting["setting"], setting
) for setting in settings])
2016-03-29 11:56:58 +00:00
if len(event["args_split"]) > 1:
setting = event["args_split"][0].lower()
if setting in settings_dict:
2016-03-29 11:56:58 +00:00
value = " ".join(event["args_split"][1:])
value = settings_dict[setting].get("validate",
lambda x: x)(value)
2016-03-29 11:56:58 +00:00
if not value == None:
if array:
current_value = target.get_setting(setting, [])
current_value.append(value)
target.set_setting(setting, current_value)
else:
target.set_setting(setting, value)
self.events.on("set").on(category).on(setting).call(
value=value, target=target)
2016-03-29 11:56:58 +00:00
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:
shown_settings = [key for key, value in settings_dict.items()
if not value.get("hidden", False)]
2016-03-29 11:56:58 +00:00
event["stdout"].write("Available settings: %s" % (
", ".join(shown_settings)))
@utils.hook("received.command.set", help="Set a specified user setting")
@utils.hook("received.command.setadd",
help="Add to a specified user setting")
2016-03-29 11:56:58 +00:00
def set(self, event):
"""
:usage: <setting> <value>
"""
self._set("set", event, event["user"], event["command"]=="setadd")
2016-03-29 11:56:58 +00:00
@utils.hook("received.command.channelset", channel_only=True,
require_mode="o", help=CHANNELSET_HELP)
@utils.hook("received.command.channelsetoverride", channel_only=True,
permission="channelsetoverride", help=CHANNELSET_HELP)
@utils.hook("received.command.channelsetadd", channel_only=True,
require_mode="o", help=CHANNELSETADD_HELP)
@utils.hook("received.command.channelsetaddoverride", channel_only=True,
permission="channelsetoverride", help=CHANNELSETADD_HELP)
2016-03-29 11:56:58 +00:00
def channel_set(self, event):
"""
:usage: <setting> <value>
"""
self._set("channelset", event, event["target"], False,
event["command"].startswith("channelsetadd"))
2018-09-09 08:34:55 +00:00
@utils.hook("received.command.serverset",
help="Set a specified server setting for the current server")
@utils.hook("received.command.serversetadd",
help="Add to a specified server setting for the current server")
2018-09-09 08:34:55 +00:00
def server_set(self, event):
"""
:usage: <setting> <value>
:permission: serverset
"""
self._set("serverset", event, event["server"],
event["command"]="serversetadd")
2018-08-08 19:45:57 +00:00
@utils.hook("received.command.botset", help="Set a specified bot setting")
@utils.hook("received.command.botsetadd",
help="Add to a specified bot setting")
def bot_set(self, event):
"""
:help: Set a specified bot setting
:usage: <setting> <value>
:permission: botset
"""
self._set("botset", event, self.bot, False,
event["command"]=="botsetadd")
2018-08-08 19:45:57 +00:00
def _get(self, event, setting, qualifier, value):
if not value == None:
if array:
value = ", ".join(value)
2018-08-08 19:45:57 +00:00
event["stdout"].write("'%s'%s: %s" % (setting,
2018-08-08 19:49:34 +00:00
qualifier, str(value)))
2016-03-29 11:56:58 +00:00
else:
2018-08-08 19:45:57 +00:00
event["stdout"].write("'%s' has no value set" % setting)
@utils.hook("received.command.get", min_args=1)
2018-09-09 08:34:55 +00:00
def get(self, event):
"""
:help: Get a specified user setting
:usage: <setting>
"""
2018-09-09 08:34:55 +00:00
setting = event["args_split"][0]
self._get(event, setting, "", event["user"].get_setting(
setting, None))
@utils.hook("received.command.channelget", channel_only=True, min_args=1)
2018-08-08 19:45:57 +00:00
def channel_get(self, event):
"""
:help: Get a specified channel setting for the current channel
:usage: <setting>
:require_mode: o
"""
2018-08-08 19:45:57 +00:00
setting = event["args_split"][0]
2018-08-08 19:49:34 +00:00
self._get(event, setting, " for %s" % event["target"].name,
event["target"].get_setting(setting, None))
2018-08-08 19:45:57 +00:00
@utils.hook("received.command.serverget", min_args=1)
2018-09-09 08:34:55 +00:00
def server_get(self, event):
"""
:help: Get a specified server setting for the current server
:usage: <setting>
:permission: serverget
"""
2018-08-08 19:45:57 +00:00
setting = event["args_split"][0]
self._get(event, setting, "", event["server"].get_setting(
setting, None))
@utils.hook("received.command.botget", min_args=1)
def bot_get(self, event):
"""
:help: Get a specified bot setting
:usage: <setting>
:permission: botget
"""
setting = event["args_split"][0]
self._get(event, setting, "", self.bot.get_setting(setting, None))