class Module(object): def __init__(self, bot, events, exports): self.bot = bot events.on("received.command.loadmodule").hook(self.load, min_args=1, permission="load-module", help="Load a module", usage="") events.on("received.command.unloadmodule").hook(self.unload, min_args=1, permission="unload-module", help="Unload a module", usage="") events.on("received.command.reloadmodule").hook(self.reload, min_args=1, permission="reload-module", help="Reload a module", usage="") events.on("received.command.enablemodule").hook(self.enable, min_args=1, permission="enable-module", help="Enable a module", usage="") events.on("received.command.disablemodule").hook(self.disable, min_args=1, permission="disable-module", help="Disable a module", usage="") def load(self, event): name = event["args_split"][0].lower() if name in self.bot.modules.modules: event["stderr"].write("Module '%s' is already loaded" % name) return self.bot.modules.load_module(name) event["stdout"].write("Loaded '%s'" % name) def unload(self, event): name = event["args_split"][0].lower() if not name in self.bot.modules.modules: event["stderr"].write("Module '%s' isn't loaded" % name) return self.bot.modules.unload_module(name) event["stdout"].write("Unloaded '%s'" % name) def reload(self, event): name = event["args_split"][0].lower() if not name in self.bot.modules.modules: event["stderr"].write("Module '%s' isn't loaded" % name) return self.bot.modules.unload_module(name) self.bot.modules.load_module(name) event["stdout"].write("Reloaded '%s'" % name) def enable(self, event): name = event["args_split"][0].lower() blacklist = self.bot.get_setting("module-blacklist", []) if not name in blacklist: event["stderr"].write("Module '%s' isn't disabled" % name) return blacklist.remove(name) event["stdout"].write("Module '%s' has been enabled and can now " "be loaded" % name) def disable(self, event): name = event["args_split"][0].lower() and_unloaded = "" if name in self.bot.modules.modules: self.bot.modules.unload_module(name) and_unloaded = " and unloaded" blacklist = self.bot.get_setting("module-blacklist", []) if name in blacklist: event["stderr"].write("Module '%s' is already disabled" % name) return blacklist.append(name) self.bot.set_setting("module-blacklist", blacklist) event["stdout"].write("Module '%s' has been disabled%s" % ( name, and_unloaded))