X-Git-Url: https://git.openstreetmap.org./osqa.git/blobdiff_plain/410bfa05ee36ed1d99356c443a5f3f6aa3ee9578..178a88bfeb53a6b9520c74c10eea42073098902b:/forum/modules/decorators.py?ds=sidebyside

diff --git a/forum/modules/decorators.py b/forum/modules/decorators.py
index 17d248b..a4c1fad 100644
--- a/forum/modules/decorators.py
+++ b/forum/modules/decorators.py
@@ -37,7 +37,7 @@ def decorate(origin, needs_origin=True):
                 return origin
             return decorator
 
-        raise Exception('Not an decoratable function: %s' % origin.name)
+        raise TypeError('Not a decoratable function: %s' % origin.__name__)
 
     def decorator(fn):
         origin.decorate(fn, needs_origin)
@@ -46,6 +46,12 @@ def decorate(origin, needs_origin=True):
     return decorator
 
 
+def decorate_all(module):
+    [setattr(module, n, decoratable(f)) for n, f in
+        [(n, getattr(module, n)) for n in dir(module)]
+        if (callable(f)) and (not inspect.isclass(f)) and (f.__module__ == module.__name__)]
+
+