exception_hook module

[erebus.git] / modlib.py

diff --git a/modlib.py b/modlib.py
index ee94e9ee38679a7dd928a89a9b69c1c40677645c..b09272e89b3704d8daa3ef3f7d03b0555db3e05e 100644 (file)
--- a/modlib.py
+++ b/modlib.py
@@ -29,16 +29,27 @@ class modlib(object):
        MANAGER =  99
        ADMIN   =  75
        STAFF   =  50
+       KNOWN   =   1
        AUTHED  =   0 # Users which have are known to be authed
        ANYONE  =  -1 # non-authed users have glevel set to -1
        IGNORED =  -2 # The default reqglevel is ANYONE, so any commands will be ignored from IGNORED users unless the command reglevel=-2
+       glevs   = {
+               'OWNER': OWNER,
+               'MANAGER': MANAGER,
+               'ADMIN': ADMIN,
+               'STAFF': STAFF,
+               'KNOWN': KNOWN,
+               'AUTHED': AUTHED,
+               'ANYONE': ANYONE,
+               'IGNORED': IGNORED,
+       }
 
        # (channel) access levels
        COWNER  =   5
        MASTER  =   4
        OP      =   3
        VOICE   =   2
-       KNOWN   =   1
+       #KNOWN  =   1 is set above by glevels
        PUBLIC  =   0 # Anyone (use glevel to control whether auth is needed)
        BANNED  =  -1 # The default reqclevel is PUBLIC, so any commands which needchan will be ignored from BANNED users unless the command reqclevel=-1
        #         [   0         1        2     3         4        5    -1]
@@ -51,6 +62,7 @@ class modlib(object):
                self.hooks = {}
                self.numhooks = {}
                self.chanhooks = {}
+               self.exceptionhooks = []
                self.helps = []
                self.parent = None
 
@@ -66,12 +78,14 @@ class modlib(object):
                #"specified" values will be printed. unspecified values will result in "OK" or "failed"
                self.parent = parent
                for cmd, func in self.hooks.items():
-                       self.parent.hook(cmd, func)
-                       self.parent.hook("%s.%s" % (self.name, cmd), func)
+                       parent.hook(cmd, func)
+                       parent.hook("%s.%s" % (self.name, cmd), func)
                for num, func in self.numhooks.items():
-                       self.parent.hooknum(num, func)
+                       parent.hooknum(num, func)
                for chan, func in self.chanhooks.items():
-                       self.parent.hookchan(chan, func)
+                       parent.hookchan(chan, func)
+               for exc, func in self.exceptionhooks:
+                       parent.hookexception(exc, func)
 
                for func, args, kwargs in self.helps:
                        try:
@@ -87,6 +101,8 @@ class modlib(object):
                        parent.unhooknum(num, func)
                for chan, func in self.chanhooks.items():
                        parent.unhookchan(chan, func)
+               for exc, func in self.exceptionhooks:
+                       parent.unhookexception(exc, func)
 
                for func, args, kwargs in self.helps:
                        try:
@@ -95,6 +111,14 @@ class modlib(object):
                                pass
                return True
 
+       def hookexception(self, exc):
+               def realhook(func):
+                       self.exceptionhooks.append((exc, func))
+                       if self.parent is not None:
+                               self.parent.hookexception(exc, func)
+                       return func
+               return realhook
+
        def hooknum(self, num):
                def realhook(func):
                        self.numhooks[str(num)] = func