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