Fix decorators

src/decorators.py

@@ -181,7 +181,10 @@ class hook:
         HOOKS[name].append(self)
 
     def __call__(self, func):
-        self.func = func
+        if isinstance(func, hook):
+            self.func = func.func
+        else:
+            self.func = func
         self.__doc__ = self.func.__doc__
         return self
 

src/handler.py

@@ -101,6 +101,8 @@ def unhandled(cli, prefix, cmd, *args):
                     notify_error(cli, botconfig.CHANNEL, log)
 
 def connect_callback(cli):
+    @hook("endofmotd", hookid=294)
+    @hook("nomotd", hookid=294)
     def prepare_stuff(cli, *args):
         # just in case we haven't managed to successfully auth yet
         if not botconfig.SASL_AUTHENTICATION:
@@ -126,9 +128,6 @@ def connect_callback(cli):
 
         wolfgame.connect_callback(cli)
 
-    hook("endofmotd", hookid=294)(prepare_stuff)
-    hook("nomotd", hookid=294)(prepare_stuff)
-
     def mustregain(cli, *blah):
         if not botconfig.PASS:
             return