mgt c7 checkpoint 1

backend/app/__init__.py

@@ -11,5 +11,5 @@ migrate = Migrate(app, db)
 login = LoginManager(app)
 login.login_view = 'login'
 
-from app import routes, models
+from app import routes, models, errors
 

backend/app/errors.py

@@ -0,0 +1,13 @@
+from flask import render_template
+
+from app import app, db
+
+@app.errorhandler(404)
+def not_found_error(error):
+    return render_template('404.html'), 404
+
+@app.errorhandler(500)
+def internal_error(error):
+    db.session.rollback()
+    return render_template('500.html'), 500
+

backend/app/templates/404.html

@@ -0,0 +1,8 @@
+{% extends "base.html" %}
+
+{% block content %}
+
+	<h1>File Not Found</h1>
+	<p><a href="{{ url_for('index') }}">Back</a></p>
+
+{% endblock %}

backend/app/templates/500.html

@@ -0,0 +1,9 @@
+{% extends "base.html" %}
+
+{% block content %}
+
+	<h1>An unexpected error has occurred.</h1>
+	<p>Administrator has been notified.</p>
+	<p><a href="{{ url_for('index') }}">Back</a></p>
+
+{% endblock %}