Restructure Views to Blueprints

labconnect/__init__.py

@@ -1,17 +1,54 @@
 import os
+import logging
 
 # Import Flask modules
 from flask import Flask
-from flask_sqlalchemy import SQLAlchemy
 
-environment = os.environ.get("CONFIG", "config.TestingConfig")
+# from flask_sqlalchemy import SQLAlchemy
+from flask_wtf.csrf import CSRFProtect
 
-# Create flask app object
-app = Flask(__name__)
-app.config.from_object(environment)
+
+csrf_protection = CSRFProtect()
 
 # Create Database object
 # db = SQLAlchemy(app)
 
-# Import all views
-import labconnect.views
+
+def create_app() -> Flask:
+    # Create flask app object
+    app = Flask(__name__)
+    app.config.from_object(os.environ.get("CONFIG", "config.TestingConfig"))
+
+    initialize_extensions(app)
+    register_blueprints(app)
+
+    return app
+
+
+# ----------------
+# Helper Functions
+# ----------------
+
+
+def initialize_extensions(app) -> None:
+    # Since the application instance is now created, pass it to each Flask
+    # extension instance to bind it to the Flask application instance (app)
+    # db.init_app(app)
+    csrf_protection.init_app(app)
+
+    # Flask-Login configuration
+    # from project.models import User
+
+    # @login.user_loader
+    # def load_user(user_id):
+    #     return User.query.filter(User.id == int(user_id)).first()
+
+
+def register_blueprints(app) -> None:
+    # Since the application instance is now created, register each Blueprint
+    # with the Flask application instance (app)
+    from labconnect.main import main_blueprint
+    from labconnect.errors import error_blueprint
+
+    app.register_blueprint(main_blueprint)
+    app.register_blueprint(error_blueprint)

labconnect/errors/__init__.py

@@ -0,0 +1,9 @@
+"""
+Blueprint for error pages
+"""
+from flask import Blueprint
+
+
+error_blueprint = Blueprint("errors", __name__, template_folder="templates")
+
+from . import routes

labconnect/errors/routes.py

@@ -0,0 +1,15 @@
+from flask import make_response, render_template, Response
+
+from . import error_blueprint
+
+
+@error_blueprint.errorhandler(404)
+def page_not_found(e) -> Response:
+    # 404 error page
+    return make_response(render_template("404.html"), 404)
+
+
+@error_blueprint.errorhandler(500)
+def error_for_server(e) -> Response:
+    # 500 error page
+    return make_response(render_template("500.html"), 500)

labconnect/main/__init__.py

@@ -0,0 +1,8 @@
+"""
+Blueprint for main pages
+"""
+from flask import Blueprint
+
+main_blueprint = Blueprint("main", __name__, template_folder="templates")
+
+from . import routes

labconnect/main/routes.py

@@ -0,0 +1,9 @@
+from flask import render_template, Response
+
+from . import main_blueprint
+
+
+@main_blueprint.route("/")
+def index():
+    # return "Hello World"
+    return render_template("home.html")

run.py

@@ -1,4 +1,6 @@
-from labconnect import app
+from labconnect import create_app
+
+app = create_app()
 
 if __name__ == "__main__":
     app.run(port=8000)