Added custom error pages.

3 files changed, 37 insertions, 0 deletions

diff --git a/pyaggr3g470r/templates/errors/404.html b/pyaggr3g470r/templates/errors/404.html
new file mode 100644
index 00000000..49c9ef5a
--- /dev/null
+++ b/pyaggr3g470r/templates/errors/404.html
@@ -0,0 +1,12 @@
+{% extends "layout.html" %}
+{% block head %}
+{{ super() }}
+{% endblock %}
+{% block content %}
+<div class="container">
+    <div class="jumbotron">
+        <h1>Page Not Found</h1>
+        <p>What you were looking for is just not there, go to the <a href="{{ url_for('home') }}">home page</a>.</p>
+    </div>
+</div>
+{% endblock %}
diff --git a/pyaggr3g470r/templates/errors/500.html b/pyaggr3g470r/templates/errors/500.html
new file mode 100644
index 00000000..1fa6acc9
--- /dev/null
+++ b/pyaggr3g470r/templates/errors/500.html
@@ -0,0 +1,12 @@
+{% extends "layout.html" %}
+{% block head %}
+{{ super() }}
+{% endblock %}
+{% block content %}
+<div class="container">
+    <div class="jumbotron">
+        <h1>Internal Server Error</h1>
+        <p>Something bad just happened! Go to the <a href="{{ url_for('home') }}">home page</a>.</p>
+    </div>
+</div>
+{% endblock %}
diff --git a/pyaggr3g470r/views.py b/pyaggr3g470r/views.py
index e526001b..c1fa459e 100644
--- a/pyaggr3g470r/views.py
+++ b/pyaggr3g470r/views.py
@@ -94,6 +94,19 @@ def load_user(email):
     # Return an instance of the User model
     return User.query.filter(User.email == email).first()
 
+
+#
+# Custom error pages.
+#
+@app.errorhandler(404)
+def page_not_found(e):
+    return render_template('errors/404.html'), 404
+
+@app.errorhandler(500)
+def page_not_found(e):
+    return render_template('errors/500.html'), 500
+
+
 def redirect_url(default='home'):
     return request.args.get('next') or \
             request.referrer or \