Allow @templated to be called raw, i.e. w/o ()

1 files changed, 11 insertions, 1 deletions

diff --git a/app/utils.py b/app/utils.py
index e6a7c95..1d5f73c 100644
--- a/app/utils.py
+++ b/app/utils.py
@@ -8,6 +8,12 @@ def _gen_tpl(endpoint):
     return endpoint.replace('.', '/') + '.jinja'
 
 def templated(template=None):
+    fun = None
+    if template is not None and callable(template):
+        # a function was passed in
+        fun = template
+        template = None
+
     def decorator(f):
         @wraps(f)
         def decorated_function(*args, **kwargs):
@@ -25,7 +31,11 @@ def templated(template=None):
                 return ctx
             return render_template(template_name, **ctx)
         return decorated_function
-    return decorator
+
+    if fun is None:
+        return decorator
+    else:
+        return decorator(fun)
 
 def redirect (target, **kwargs):
     code = kwargs.pop('_code', None)