From 52ffd35abc47abf5bdaa8d68c0947fb650849b85 Mon Sep 17 00:00:00 2001
From: Ben Kurtovic <ben.kurtovic@gmail.com>
Date: Wed, 21 Dec 2016 00:17:39 -0500
Subject: [PATCH] Add custom 404.

---
 app.py             | 4 ++++
 templates/404.mako | 6 ++++++
 2 files changed, 10 insertions(+)
 create mode 100644 templates/404.mako

diff --git a/app.py b/app.py
index 307e0dd..3fdb4db 100755
--- a/app.py
+++ b/app.py
@@ -84,5 +84,9 @@ def set_style(style):
         abort(404)
     return "", 204
 
+@app.errorhandler(404)
+def page_not_found(err):
+    return render_template("404.mako"), 404
+
 if __name__ == "__main__":
     app.run(debug=True, port=8080)
diff --git a/templates/404.mako b/templates/404.mako
new file mode 100644
index 0000000..8096f25
--- /dev/null
+++ b/templates/404.mako
@@ -0,0 +1,6 @@
+<%inherit file="_base.mako"/>
+<%block name="title">
+    404 &ndash; ${g.config.get("corp.name") | h}
+</%block>
+<h1>Error 404</h1>
+<p>The page you asked for couldn't be found.</p>