cwinfo/powerdns-admin
5
0
Fork 0
mirror of https://github.com/cwinfo/powerdns-admin.git synced 2024-11-09 23:20:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #406 from ymage/fix-403

Browse Source 
Make a redirection to previous page after logged in
This commit is contained in:
Khanh Ngo 2018-11-26 10:35:43 +07:00 committed by GitHub
commit 32c442e083
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
app/views.py
View File

@ -299,8 +299,13 @@ def saml_authorized():
return render_template('errors/SAML.html', errors=errors) return render_template('errors/SAML.html', errors=errors)
@app.route('/login', methods=['GET', 'POST'])
@login_manager.unauthorized_handler @login_manager.unauthorized_handler
def unauthorized_callback():
session['next'] = request.path
return redirect('/login')
@app.route('/login', methods=['GET', 'POST'])
def login(): def login():
SAML_ENABLED = app.config.get('SAML_ENABLED') SAML_ENABLED = app.config.get('SAML_ENABLED')
@ -425,7 +430,7 @@ def login():
return render_template('login.html', saml_enabled=SAML_ENABLED, error='Token required') return render_template('login.html', saml_enabled=SAML_ENABLED, error='Token required')
login_user(user, remember = remember_me) login_user(user, remember = remember_me)
return redirect(request.args.get('next') or url_for('index')) return redirect(session.get('next', url_for('index')))
else: else:
if not username or not password or not email: if not username or not password or not email:
return render_template('register.html', error='Please input required information') return render_template('register.html', error='Please input required information')