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'))