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", @utils.export("channelset", {"setting": "github-hook",
"help": ("Disable/Enable showing BitBot's github commits in the " "help": ("Disable/Enable showing BitBot's github commits in the "
"current channel"), "validate": utils.bool_or_none, "current channel"), "hidden": True})
"hidden": True})
class Module(ModuleManager.BaseModule): class Module(ModuleManager.BaseModule):
@utils.hook("api.post.github") @utils.hook("api.post.github")
def github(self, event): def github(self, event):

49
modules/set.py
View file

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