on user login, redirect based on next query parameter if available

atst/routes/__init__.py

@@ -80,6 +80,13 @@ def _make_authentication_context():
     )
 
 
+def redirect_url():
+    if request.args.get("next"):
+        return request.args.get("next")
+    else:
+        return url_for(".home")
+
+
 @bp.route("/login-redirect")
 def login_redirect():
     auth_context = _make_authentication_context()
@@ -87,7 +94,7 @@ def login_redirect():
     user = auth_context.get_user()
     session["user_id"] = user.id
 
-    return redirect(url_for(".home"))
+    return redirect(redirect_url())
 
 
 @bp.route("/logout")

atst/routes/dev.py

@@ -1,5 +1,6 @@
 from flask import Blueprint, request, session, redirect, url_for
 
+from . import redirect_url
 from atst.domain.users import Users
 
 bp = Blueprint("dev", __name__)
@@ -63,4 +64,4 @@ def login_dev():
     )
     session["user_id"] = user.id
 
-    return redirect(url_for("atst.home"))
+    return redirect(redirect_url())