1 files changed, 25 insertions, 0 deletions
diff --git a/utils/request.py b/utils/request.py
new file mode 100644
index 0000000..8d36ed6
--- /dev/null
+++ b/utils/request.py
@@ -0,0 +1,25 @@
+from flask import g, request
+
+from app import app
+
+
+def after_this_request(f):
+    """
+    Decorator to execute methods after the request is handled, to
+    modify the response before sending back to the client. This could
+    be used to set cookies.
+    """
+
+    if not hasattr(g, 'after_request_callbacks'):
+        g.after_request_callbacks = []
+
+    g.after_request_callbacks.append(f)
+    return f
+
+
+@app.after_request
+def call_after_request_callbacks(response):
+    for callback in getattr(g, 'after_request_callbacks', ()):
+        callback(response)
+
+    return response