search module, wip

realms/lib/hook.py

@@ -7,12 +7,15 @@ def hook_func(name, fn):
     @wraps(fn)
     def wrapper(self, *args, **kwargs):
         for hook, a, kw in self.__class__._pre_hooks.get(name) or []:
-            hook(*a, **kw)
+            hook(*args, **kwargs)
 
         rv = fn(self, *args, **kwargs)
 
+        # Attach return value for post hooks
+        kwargs.update(dict(rv=rv))
+
         for hook, a, kw in self.__class__._post_hooks.get(name) or []:
-            hook(*a, **kw)
+            hook(*args, **kwargs)
 
         return rv
     return wrapper