utils: add templated decorator from flask patterns

1 files changed, 20 insertions, 0 deletions

diff --git a/utils.py b/utils.py
new file mode 100644
index 0000000..147f048
--- /dev/null
+++ b/utils.py
@@ -0,0 +1,20 @@
+from functools import wraps
+from flask import request, render_template
+
+# using http://flask.pocoo.org/docs/patterns/viewdecorators/
+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('.', '/') + '.html'
+            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