Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Support arrays of user/channel/bot/server settings

Browse source
This commit is contained in:
jesopo 2018-11-06 13:01:30 +00:00
parent 28a00e01de
commit 2921ed77ae
2 changed files with 38 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
modules/github.py
View file

@ -5,8 +5,7 @@ COMMIT_URL = "https://github.com/%s/commit/%s"
@utils.export("channelset", {"setting": "github-hook",
"help": ("Disable/Enable showing BitBot's github commits in the "
"current channel"), "validate": utils.bool_or_none,
"hidden": True})
"current channel"), "hidden": True})
class Module(ModuleManager.BaseModule):
@utils.hook("api.post.github")
def github(self, event):

51
modules/set.py
View file

@ -1,7 +1,11 @@
from src import ModuleManager, utils
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):
def _set(self, category, event, target, array):
settings = self.exports.get_all(category)
settings_dict = dict([(setting["setting"], setting
) for setting in settings])
@ -14,7 +18,13 @@ class Module(ModuleManager.BaseModule):
lambda x: x)(value)
if not value == None:
target.set_setting(setting, value)
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)
@ -31,45 +41,58 @@ class Module(ModuleManager.BaseModule):
event["stdout"].write("Available settings: %s" % (
", ".join(shown_settings)))
@utils.hook("received.command.set")
@utils.hook("received.command.set", help="Set a specified user setting")
@utils.hook("received.command.setadd",
help="Add to a specified user setting")
def set(self, event):
"""
:help: Set a specified user setting
:usage: <setting> <value>
"""
self._set("set", event, event["user"])
self._set("set", event, event["user"], event["command"]=="setadd")
@utils.hook("received.command.channelset", channel_only=True,
require_mode="o")
require_mode="o", help=CHANNELSET_HELP)
@utils.hook("received.command.channelsetoverride", channel_only=True,
permission="channelsetoverride")
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)
def channel_set(self, event):
"""
:help: Get a specified channel setting for the current channel
:usage: <setting> <value>
"""
self._set("channelset", event, event["target"])
self._set("channelset", event, event["target"], False,
event["command"].startswith("channelsetadd"))
@utils.hook("received.command.serverset")
@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")
def server_set(self, event):
"""
:help: Set a specified server setting for the current server
:usage: <setting> <value>
:permission: serverset
"""
self._set("serverset", event, event["server"])
self._set("serverset", event, event["server"],
event["command"]="serversetadd")
@utils.hook("received.command.botset")
@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)
self._set("botset", event, self.bot, False,
event["command"]=="botsetadd")
def _get(self, event, setting, qualifier, value):
if not value == None:
if array:
value = ", ".join(value)
event["stdout"].write("'%s'%s: %s" % (setting,
qualifier, str(value)))
else: