smoothstate: honor redirects (merged pr 267)

1 files changed, 4 insertions, 1 deletions

diff --git a/app.py b/app.py
index 29e5543..7f835ca 100644
--- a/app.py
+++ b/app.py
@@ -7,6 +7,7 @@ from config import config
 from flask import (
     Flask,
     render_template,
+    make_response,
     Markup,
     request,
     g,
@@ -302,7 +303,9 @@ def page(num):
     if 'userlogin' in session:
         user['loggedin'] = True
         user['name'] = session['userlogin']
-    return render_template('index.html', currentpage=num, pages=[[]]*numpages, jokes=jokes, user=user)
+    r = make_response(render_template('index.html', currentpage=num, pages=[[]]*numpages, jokes=jokes, user=user))
+    r.headers.set('X-SmoothState-Location', request.path)
+    return r
 
 @app.route('/submit', methods=['POST'])
 def submit():