diff options
Diffstat (limited to '')
-rw-r--r-- | app/utils.py | 12 |
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) |