X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/69071d33cc3db3ecc258b945dd7a9d9d4d92b391..b9c6eb1d457efe531333ab32818ca5de2bc696e3:/modlib.py diff --git a/modlib.py b/modlib.py index d2fe492..43187f9 100644 --- a/modlib.py +++ b/modlib.py @@ -60,9 +60,9 @@ class modlib(object): def hooknum(self, num): def realhook(func): - self.numhooks[num] = func + self.numhooks[str(num)] = func if self.parent is not None: - self.parent.hooknum(num, func) + self.parent.hooknum(str(num), func) return func return realhook @@ -74,8 +74,13 @@ class modlib(object): return func return realhook - def hook(self, cmd, needchan=True, glevel=ANYONE, clevel=PUBLIC): + def hook(self, cmd=None, needchan=True, glevel=ANYONE, clevel=PUBLIC): + _cmd = cmd #save this since it gets wiped out... def realhook(func): + cmd = _cmd #...and restore it + if cmd is None: + cmd = func.__name__ # default to function name + func.needchan = needchan func.reqglevel = glevel func.reqclevel = clevel @@ -86,6 +91,12 @@ class modlib(object): return func return realhook + def mod(self, modname): + if self.parent is not None: + return self.parent.module(modname) + else: + return error('unknown parent') + def argsEQ(self, num): def realhook(func): def checkargs(bot, user, chan, realtarget, *args): @@ -93,6 +104,7 @@ class modlib(object): return func(bot, user, chan, realtarget, *args) else: bot.msg(user, self.WRONGARGS) + checkargs.__name__ = func.__name__ return checkargs return realhook @@ -103,5 +115,22 @@ class modlib(object): return func(bot, user, chan, realtarget, *args) else: bot.msg(user, self.WRONGARGS) + checkargs.__name__ = func.__name__ return checkargs return realhook + + def help(*args, **kwargs): + """help(args, shorthelp, longhelp, more lines longhelp, cmd=...?) + Example: + help(" ", "login") + ^ Help will only be one line. Command name determined based on function name. + help(" ", "add a user", cmd=("adduser", "useradd")) + ^ Help will be listed under ADDUSER; USERADD will say "alias for adduser" + help(None, "do stuff", "This command is really complicated.") + ^ Command takes no args. Short description (in overall HELP listing) is "do stuff". + Long description (HELP ) will say " - do stuff", newline, "This command is really complicated." + """ + try: + self.mod('help').reghelp(*args, **kwargs) + except: + pass