* - change old code to use newer cfg.getboolean instead of bool(int())

[erebus.git] / modlib.py

diff --git a/modlib.py b/modlib.py
index 9f59573eb6a9e7e20d4f83f4a5c2426c666edad2..ff63a07612a9fd0fcf6157bb820275a7d635ace6 100644 (file)
--- a/modlib.py
+++ b/modlib.py
@@ -30,6 +30,8 @@ class modlib(object):
        KNOWN   =   1
        PUBLIC  =   0 #anyone (use glevel to control auth-needed)
        BANNED  =  -1
+       #         [   0         1        2     3         4        5    -1]
+       clevs   = [None, 'Friend', 'Voice', 'Op', 'Master', 'Owner', None]
 
        # messages
        WRONGARGS = "Wrong number of arguments."
@@ -98,7 +100,8 @@ class modlib(object):
                        return func
                return realhook
 
-       def hook(self, cmd=None, needchan=True, glevel=ANYONE, clevel=PUBLIC):
+       def hook(self, cmd=None, needchan=True, glevel=ANYONE, clevel=PUBLIC, wantchan=None):
+               if wantchan is None: wantchan = needchan
                _cmd = cmd #save this since it gets wiped out...
                def realhook(func):
                        cmd = _cmd #...and restore it
@@ -108,6 +111,7 @@ class modlib(object):
                                cmd = (cmd,)
 
                        func.needchan = needchan
+                       func.wantchan = wantchan
                        func.reqglevel = glevel
                        func.reqclevel = clevel
                        func.cmd = cmd