refactor for healthz endpoint

This commit is contained in:
2025-01-12 09:23:38 -07:00
parent 2b1023fbd3
commit 666570b069
3 changed files with 33 additions and 12 deletions

View File

@@ -1,7 +1,5 @@
from flask import Flask, g, request
from sqlalchemy import select
from flask import Flask
from . import db as dbm
from .config import Config
from .ext import db
from .v1_api import bp as v1_bp
@@ -13,14 +11,6 @@ def create_app():
db.init_app(app)
@app.before_request
def before_request():
if not hasattr(g, "tenant"):
hostname = request.host.split(":")[0]
g.tenant = db.session.scalars(
select(dbm.Tenant).filter_by(hostname=hostname).limit(1)
).first()
@app.teardown_request
def teardown_request(exc):
if exc:
@@ -29,4 +19,8 @@ def create_app():
app.register_blueprint(v1_bp)
@app.route("/healthz")
def healthz():
return {"status": "ok"}
return app

View File

@@ -1,9 +1,23 @@
from flask import Blueprint
from flask import Blueprint, Flask, g, request
from flask_restful import Api
from sqlalchemy import select
from ..ext import db
from . import db as dbm
from .flights import FlightCollectionResource, FlightResource
bp = Blueprint("api", __name__, url_prefix="/api")
@bp.before_request
def before_request():
if not hasattr(g, "tenant"):
hostname = request.host.split(":")[0]
g.tenant = db.session.scalars(
select(dbm.Tenant).filter_by(hostname=hostname).limit(1)
).first()
api = Api(bp)
api.add_resource(FlightCollectionResource, "/v1/flights")