From afac3b628144caa2225801c5442d779e397e54d0 Mon Sep 17 00:00:00 2001 From: Blake Ridgway Date: Thu, 3 Apr 2025 10:38:51 -0500 Subject: [PATCH] refactor: Improve server setup and add health check Added /health endpoint for monitoring Improved readability and config setup Imported auth_bp directly --- server.py | 25 ++++++++++++++++--------- 1 file changed, 16 insertions(+), 9 deletions(-) diff --git a/server.py b/server.py index c952050..1b00247 100644 --- a/server.py +++ b/server.py @@ -1,25 +1,32 @@ import os from flask import Flask -from models import db, init_db -from routes.user_auth.auth import auth_bp -from dotenv import load_dotenv from flask_cors import CORS +from dotenv import load_dotenv + +from models import db, init_db +from routes.user_auth import auth load_dotenv() app = Flask(__name__) -CORS(app) +app.config["SECRET_KEY"] = os.getenv("SECRET_KEY") +app.config["SQLALCHEMY_DATABASE_URI"] = os.getenv("DATABASE") +app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False -app.secret_key = os.getenv('SECRET_KEY') -app.config['SQLALCHEMY_DATABASE_URI'] = os.getenv('DATABASE') -app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False +CORS(app) # Consider specific origins in production init_db(app) +app.register_blueprint(auth.auth_bp) + + +@app.route("/health") +def health_check(): + """Health check endpoint.""" + return "OK", 200 -app.register_blueprint(auth_bp) with app.app_context(): db.create_all() -if __name__ == '__main__': +if __name__ == "__main__": app.run(debug=True)