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

This commit is contained in:
dandds
2018-10-01 14:22:44 -04:00
parent 9f4b284d6c
commit 8a89c519eb
3 changed files with 25 additions and 4 deletions

View File

@@ -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")

View File

@@ -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())