From da81ea21bc14446548204bf77633d3248ea9bc48 Mon Sep 17 00:00:00 2001 From: Noah Petherbridge Date: Fri, 9 May 2014 19:29:49 +0000 Subject: [PATCH] Add url param to login form to control redirect Site layouts that include a login form on every page can now include their relative URI (`request.path`) so the user can be redirected back to the same page after login. --- rophako/modules/account.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/rophako/modules/account.py b/rophako/modules/account.py index 90b8403..d60c7f8 100644 --- a/rophako/modules/account.py +++ b/rophako/modules/account.py @@ -34,6 +34,12 @@ def login(): session["uid"] = db["uid"] session["name"] = db["name"] session["role"] = db["role"] + + # Redirect them to a local page? + url = request.form.get("url", "") + if url[0] == "/": + return redirect(url) + return redirect(url_for("index")) else: flash("Authentication failed.") @@ -126,4 +132,4 @@ def validate_create_form(username, pw1=None, pw2=None, skip_passwd=False): if len(errors): return errors else: - return None \ No newline at end of file + return None