Implement !botset/!botget in modules/set.py

2 files changed, 22 insertions, 0 deletions

diff --git a/modules/rest_api.py b/modules/rest_api.py
index fad3f6be..2522b109 100644
--- a/modules/rest_api.py
+++ b/modules/rest_api.py
@@ -36,6 +36,9 @@ class Handler(http.server.BaseHTTPRequestHandler):
             pass
         _bot.lock.release()
 
+@utils.export("botset", {"setting": "rest-api",
+    "help": "Enable/disable REST API",
+    "validate": utils.bool_or_none})
 class Module(object):
     def __init__(self, bot, events, exports):
         self.bot = bot
diff --git a/modules/set.py b/modules/set.py
index 3c8bd332..3e5b264f 100644
--- a/modules/set.py
+++ b/modules/set.py
@@ -50,6 +50,15 @@ class Module(ModuleManager.BaseModule):
         """
         self._set(self.exports.get_all("serverset"), event, event["server"])
 
+    @utils.hook("received.command.botset")
+    def bot_set(self, event):
+        """
+        :help: Set a specified bot setting
+        :usage: <setting> <value>
+        :permission: botset
+        """
+        self._set(self.exports.get_all("botset"), event, self.bot)
+
     def _get(self, event, setting, qualifier, value):
         if not value == None:
             event["stdout"].write("'%s'%s: %s" % (setting,
@@ -88,3 +97,13 @@ class Module(ModuleManager.BaseModule):
         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))