Store next url redirect for in session to work with oauth login callbacks

This commit is contained in:
Chase Sterling 2016-07-08 23:39:11 -04:00
parent 0b931d1a05
commit aa0a8a2aa8
4 changed files with 8 additions and 6 deletions

View file

@ -1,4 +1,4 @@
from flask import current_app, request, redirect, Blueprint, flash, url_for
from flask import current_app, request, redirect, Blueprint, flash, url_for, session
from ..ldap.models import User
from flask_ldap_login import LDAPLoginForm
@ -14,6 +14,6 @@ def login():
return redirect(url_for('auth.login'))
if User.auth(form.user, request.form['password']):
return redirect(request.args.get("next") or url_for(current_app.config['ROOT_ENDPOINT']))
return redirect(session.get("next_url") or url_for(current_app.config['ROOT_ENDPOINT']))
else:
return redirect(url_for('auth.login'))