addde message template and 401 handler

2 files changed, 23 insertions, 2 deletions

diff --git a/app/templates/message.html b/app/templates/message.html
new file mode 100644
index 0000000..24603ef
--- /dev/null
+++ b/app/templates/message.html
@@ -0,0 +1,17 @@
+{% extends "base.html" %}
+{% block title %}Message{% endblock %}
+{% block head %}
+    {{ super() }}
+{% endblock %}
+{% block content %}
+<section class="py-5 text-center container">
+        <div class="col-lg-6 col-md-8 mx-auto">
+            {% if code is defined %}
+        <h1 class="display-1">{{code}}</h1>
+            {% endif %}
+            <p class="lead text-muted">{{ message }}</p>
+        
+        <p>Click <a href={{url_for("index")}}>here</a> to go home.</p>
+    </div>
+</section>
+{% endblock %}
\ No newline at end of file
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)