From 6fa0ea598cb629db45b6a99d83860a5f4846c3e8 Mon Sep 17 00:00:00 2001 From: Robbe Van Herck Date: Thu, 29 Nov 2018 16:14:20 +0100 Subject: [PATCH] Added redirect on 401 Unauthorized --- login.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/login.py b/login.py index 28ebafa..368e32c 100644 --- a/login.py +++ b/login.py @@ -13,6 +13,10 @@ def load_user(userid): return User.query.filter_by(id=userid).first() +@app.errorhandler(401) +def custom_401(_): + return redirect(url_for('login')) + @app.route('/login') def login(): return zeus_login() @@ -23,4 +27,4 @@ def logout(): if 'zeus_token' in session: session.pop('zeus_token', None) logout_user() - return redirect(url_for('all_users')) \ No newline at end of file + return redirect(url_for('all_users'))