Add default 404 error page. (#66)

artefactual-labs · Sep 8, 2023 · f72d8fb · f72d8fb
1 parent a315d91
commit f72d8fb
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/AIPscan/__init__.py b/AIPscan/__init__.py
@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 
-from flask import Flask
+from flask import Flask, render_template
 from flask_sqlalchemy import SQLAlchemy
 
 from AIPscan.celery import configure_celery
@@ -34,4 +34,8 @@ def create_app(config_name="default"):
 
         db.create_all()
 
+        @app.errorhandler(404)
+        def page_not_found(e):
+            return render_template("error/404.html"), 404
+
         return app
diff --git a/AIPscan/templates/error/404.html b/AIPscan/templates/error/404.html
@@ -0,0 +1,14 @@
+{% extends "base.html" %}
+
+{% block content %}
+
+<div class="row">
+    <div class="col-md-12 d-flex flex-column justify-content-center align-items-center vh-100">
+        <h1>404</h1>
+        <h4>Page not found</h4>
+	<p>The page you are looking for does not exist.</p>
+	<a href="{{ url_for('aggregator.ss_default') }}">Back To Home</a>
+    </div>
+</div>
+
+{% endblock %}