X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/f164fd1c3b740a4e926e5f8633f51b5a952c185d..dcc5bde3beb88310fdc2815383ba1509e3997607:/modlib.py

diff --git a/modlib.py b/modlib.py
index 7225ac7..ff63a07 100644
--- a/modlib.py
+++ b/modlib.py
@@ -100,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
@@ -110,6 +111,7 @@ class modlib(object):
 				cmd = (cmd,)
 
 			func.needchan = needchan
+			func.wantchan = wantchan
 			func.reqglevel = glevel
 			func.reqclevel = clevel
 			func.cmd = cmd