refactor

2 files changed, 23 insertions, 0 deletions

diff --git a/app/errors/__init__.py b/app/errors/__init__.py
new file mode 100644
index 0000000..8a85dca
--- /dev/null
+++ b/app/errors/__init__.py
@@ -0,0 +1,7 @@
+# -*- coding: utf-8 -*-
+
+from flask import Blueprint
+
+bp = Blueprint("errors", __name__)
+
+from app.errors import handlers
diff --git a/app/errors/handlers.py b/app/errors/handlers.py
new file mode 100644
index 0000000..6c2b1e7
--- /dev/null
+++ b/app/errors/handlers.py
@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+
+from flask import render_template
+from app import db
+from app.errors import bp
+
+
+@bp.app_errorhandler(404)
+def not_found_error(error):
+    return render_template("errors/404.html"), 404
+
+
+@bp.app_errorhandler(500)
+def internal_error(error):
+    db.session.rollback()
+    return render_template("errors/500.html"), 500