Implement basic error page

1 files changed, 9 insertions, 4 deletions

diff --git a/app.py b/app.py
index 7d93b4f..353de94 100644
--- a/app.py
+++ b/app.py
@@ -70,7 +70,7 @@ def context_processor():
 
 @app.before_request
 def before_request():
-    if request.endpoint == "static":
+    if not request.endpoint or request.endpoint == "static":
         return
     db = get_db()
     cur = db.cursor()
@@ -104,9 +104,14 @@ def before_request():
 
 
 @app.errorhandler(HTTPException)
-def handle_bad_request(e):
-    # TODO: Create an actual error page.
-    return e
+def handle_bad_request(error):
+    if error.code == 501:
+        if "username" in session:
+            flash("This content requires additional permissions to view.")
+        else:
+            flash("You must be logged in to access this content.")
+        return redirect(url_for("login"))
+    return render_template("error.html", error=error), error.code
 
 
 @app.route("/")