diff options
Diffstat (limited to 'app/views')
| -rw-r--r-- | app/views/auth.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/app/views/auth.py b/app/views/auth.py index a915cf4..1d1cafb 100644 --- a/app/views/auth.py +++ b/app/views/auth.py @@ -42,12 +42,16 @@ def signin_user(): flash_errors(form) return render_template("auth/signin.html",form=form) -@flask_login.login_required @app.route("/dashboard") +@flask_login.login_required def user_dashboard(): return render_template("dashboard.html",user=flask_login.current_user) @app.route('/logout') def logout(): flask_login.logout_user() - return 'Logged out' + return render_template("message.html",message="You have been logged out.") + +@login_manager.unauthorized_handler +def unauthorized(): + return render_template("message.html",message="You need to be logged in to access this resource", code=401) |