diff options
Diffstat (limited to '')
-rw-r--r-- | app/utils.py | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/app/utils.py b/app/utils.py new file mode 100644 index 0000000..caacea7 --- /dev/null +++ b/app/utils.py @@ -0,0 +1,29 @@ +from functools import wraps +from flask import request, render_template, url_for +from flask import redirect as _redirect + +def templated(template=None): + def decorator(f): + @wraps(f) + def decorated_function(*args, **kwargs): + template_name = template + if template_name is None: + template_name = request.endpoint \ + .replace('.', '/') + '.jinja' + ctx = f(*args, **kwargs) + if ctx is None: + ctx = {} + elif not isinstance(ctx, dict): + return ctx + return render_template(template_name, **ctx) + return decorated_function + return decorator + +def redirect (target, **kwargs): + code = kwargs.pop("_code", None) + url = url_for(target, **kwargs) + + if code is None: + return _redirect(url) + else: + return _redirect(url, code) |