Better communicate issues with load/unload/reload of modules
This commit is contained in:
jesopo
parent
7522b0fae2
commit
b7f7c9342b
2 changed files with 65 additions and 42 deletions
|
|
@ -2,6 +2,23 @@ import glob, imp, inspect, os, sys, uuid
|
|||
|
||||
BITBOT_HOOKS_MAGIC = "__bitbot_hooks"
|
||||
|
||||
class ModuleException(Exception):
|
||||
pass
|
||||
class ModuleWarning(Exception):
|
||||
pass
|
||||
|
||||
class ModuleNotFoundException(ModuleException):
|
||||
pass
|
||||
class ModuleNameCollisionException(ModuleException):
|
||||
pass
|
||||
class ModuleLoadException(ModuleException):
|
||||
pass
|
||||
class ModuleUnloadException(ModuleException):
|
||||
pass
|
||||
|
||||
class ModuleNotLoadedWarning(ModuleWarning):
|
||||
pass
|
||||
|
||||
class BaseModule(object):
|
||||
def __init__(self, bot, events, exports):
|
||||
pass
|
||||
|
|
@ -33,28 +50,32 @@ class ModuleManager(object):
|
|||
# this is a hashflag
|
||||
if line == "#--ignore":
|
||||
# nope, ignore this module.
|
||||
return None
|
||||
raise ModuleNotLoadedWarning("module ignored")
|
||||
elif line_split[0] == "#--require-config" and len(
|
||||
line_split) > 1:
|
||||
if not line_split[1].lower() in self.bot.config or not self.bot.config[
|
||||
line_split[1].lower()]:
|
||||
# nope, required config option not present.
|
||||
return None
|
||||
raise ModuleNotLoadedWarning(
|
||||
"required config not present")
|
||||
elif line_split[0] == "#--require-module" and len(
|
||||
line_split) > 1:
|
||||
if not "bitbot_%s" % line_split[1].lower() in sys.modules:
|
||||
if not line_split[1].lower() in self.waiting_requirement:
|
||||
self.waiting_requirement[line_split[1].lower()] = set([])
|
||||
self.waiting_requirement[line_split[1].lower()].add(path)
|
||||
return None
|
||||
raise ModuleNotLoadedWarning(
|
||||
"waiting for requirement")
|
||||
else:
|
||||
break
|
||||
module = imp.load_source(name, path)
|
||||
|
||||
if not hasattr(module, "Module"):
|
||||
raise ImportError("module '%s' doesn't have a Module class.")
|
||||
raise ModuleLoadException("module '%s' doesn't have a "
|
||||
"'Module' class.")
|
||||
if not inspect.isclass(module.Module):
|
||||
raise ImportError("module '%s' has a Module attribute but it is not a class.")
|
||||
raise ModuleLoadException("module '%s' has a 'Module' attribute "
|
||||
"but it is not a class.")
|
||||
|
||||
context = str(uuid.uuid4())
|
||||
context_events = self.events.new_context(context)
|
||||
|
|
@ -83,26 +104,32 @@ class ModuleManager(object):
|
|||
def load_module(self, name):
|
||||
try:
|
||||
module = self._load_module(name)
|
||||
except ImportError as e:
|
||||
self.bot.log.error("failed to load module \"%s\": %s",
|
||||
except ModuleWarning as warning:
|
||||
self.bot.log.error("Module '%s' not loaded", [name])
|
||||
raise
|
||||
except Exception as e:
|
||||
self.bot.log.error("Failed to load module \"%s\": %s",
|
||||
[name, e.msg])
|
||||
return
|
||||
if module:
|
||||
raise
|
||||
|
||||
self.modules[module._import_name] = module
|
||||
if name in self.waiting_requirement:
|
||||
for requirement_name in self.waiting_requirement:
|
||||
self.load_module(requirement_name)
|
||||
self.bot.log.info("Module '%s' loaded", [name])
|
||||
else:
|
||||
self.bot.log.error("Module '%s' not loaded", [name])
|
||||
|
||||
def load_modules(self, whitelist=[], blacklist=[]):
|
||||
for path in self.list_modules():
|
||||
name = self._module_name(path)
|
||||
if name in whitelist or (not whitelist and not name in blacklist):
|
||||
try:
|
||||
self.load_module(name)
|
||||
except ModuleWarning:
|
||||
pass
|
||||
|
||||
def unload_module(self, name):
|
||||
if not name in self.modules:
|
||||
raise ModuleNotFoundException()
|
||||
module = self.modules[name]
|
||||
del self.modules[name]
|
||||
|
||||
|
|
|
|||
|
|
@ -1,10 +1,8 @@
|
|||
|
||||
import ModuleManager
|
||||
|
||||
class Module(object):
|
||||
def __init__(self, bot, events, exports):
|
||||
self.bot = bot
|
||||
self.module_name = False
|
||||
self.silent = False
|
||||
|
||||
events.on("received.command.loadmodule").hook(self.load,
|
||||
min_args=1, permission="load-module", help="Load a module",
|
||||
|
|
@ -43,37 +41,35 @@ class Module(object):
|
|||
self.bot.modules.unload_module(name)
|
||||
event["stdout"].write("Unloaded '%s'" % name)
|
||||
|
||||
def reload(self, event):
|
||||
name = self.module_name if self.module_name != False else event[
|
||||
"args_split"][0].lower()
|
||||
if not name in self.bot.modules.modules:
|
||||
if self.silent == False:
|
||||
event["stderr"].write("Module '%s' isn't loaded" % name)
|
||||
return
|
||||
def _reload(self, name):
|
||||
self.bot.modules.unload_module(name)
|
||||
self.bot.modules.load_module(name)
|
||||
|
||||
if self.silent == False:
|
||||
def reload(self, event):
|
||||
name = event["args_split"][0].lower()
|
||||
try:
|
||||
self._reload(name)
|
||||
except ModuleManager.ModuleNotFoundException:
|
||||
event["stderr"].write("Module '%s' isn't loaded" % name)
|
||||
return
|
||||
event["stdout"].write("Reloaded '%s'" % name)
|
||||
|
||||
def reload_all(self, event):
|
||||
modules_reloaded = []
|
||||
self.silent = True
|
||||
|
||||
for name, value in self.bot.modules.modules.items():
|
||||
if name in modules_reloaded:
|
||||
pass
|
||||
|
||||
self.module_name = name
|
||||
self.reload(event)
|
||||
modules_reloaded.append(name)
|
||||
reloaded = []
|
||||
failed = []
|
||||
for name in self.bot.modules.modules.keys():
|
||||
try:
|
||||
self._reload(name)
|
||||
except ModuleManager.ModuleNotFoundException:
|
||||
failed.append
|
||||
if not self._reload(event):
|
||||
failed.append(name)
|
||||
else:
|
||||
reloaded.append(name)
|
||||
|
||||
event["stdout"].write("Reloaded modules: %s" % \
|
||||
" ".join(modules_reloaded))
|
||||
|
||||
self.silent = False
|
||||
self.module_name = False
|
||||
|
||||
def enable(self, event):
|
||||
name = event["args_split"][0].lower()
|
||||
blacklist = self.bot.get_setting("module-blacklist", [])
|
||||
|
|
|
|||
Loading…
Reference in a new issue