X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/8ba566068fc22454dc1ec3d296b85f35aa7aec17..5f42c250f797d8cf1bca89d431134764d97c63fd:/ctlmod.py diff --git a/ctlmod.py b/ctlmod.py index 8cfa7d0..ab9145a 100644 --- a/ctlmod.py +++ b/ctlmod.py @@ -1,7 +1,7 @@ # Erebus IRC bot - Author: John Runyon # module loading/unloading/tracking code -import sys +import sys, time import modlib modules = {} @@ -10,14 +10,35 @@ dependents = {} def isloaded(modname): return modname in modules def modhas(modname, attname): return getattr(modules[modname], attname, None) is not None -def load(parent, modname): +def load(parent, modname, dependent=False): + #wrapper to call _load and print return + if dependent: + print "Loading dependency %s..." % (modname), + else: + print "%05.3f [MOD] [#] Loading %s... " % (time.time() % 100000, modname), + modstatus = _load(parent, modname, dependent) + if not modstatus: + print str(modstatus) + elif modstatus == True: + if dependent: + print "OK. ", + else: + print "OK." + else: + print modstatus + return modstatus + +def _load(parent, modname, dependent=False): if not isloaded(modname): + sys.path.insert(0, 'modules') try: mod = __import__(modname) reload(mod) #in case it's been previously loaded. except BaseException as e: #we don't want even sys.exit() to crash us (in case of malicious module) so use BaseException return modlib.error(e) - + finally: + del sys.path[0] #remove ./modules from path, in case there's a name conflict + if not hasattr(mod, 'modinfo'): return modlib.error('no modinfo') @@ -30,7 +51,7 @@ def load(parent, modname): for dep in mod.modinfo['depends']: if dep not in modules: - depret = load(parent, dep) + depret = load(parent, dep, dependent=True) if not depret: return dependents[dep].append(modname)