Add alembic version endpoint

discovery-provider/src/queries/get_alembic_version.py

@@ -0,0 +1,24 @@
+from typing import TypedDict
+
+import sqlalchemy
+from src.utils import db_session
+
+
+class AlembicVersion(TypedDict):
+    version_num: str
+
+
+def get_alembic_version() -> AlembicVersion:
+    """
+    Fetches the alembic version at head from the database
+    """
+    db = db_session.get_db_read_replica()
+    with db.scoped_session() as session:
+        version = session.execute(
+            sqlalchemy.text(
+                """
+                SELECT * FROM "alembic_version";
+                """
+            )
+        ).first()
+        return dict(version)

discovery-provider/src/queries/health_check.py

@@ -3,6 +3,7 @@
 
 from flask import Blueprint, request
 from src.api_helpers import success_response
+from src.queries.get_alembic_version import get_alembic_version
 from src.queries.get_health import get_health, get_latest_ipld_indexed_block
 from src.queries.get_latest_play import get_latest_play
 from src.queries.get_sol_plays import get_latest_sol_play_check_info
@@ -21,6 +22,12 @@ def version():
     return success_response(disc_prov_version, sign_response=False)
 
 
+@bp.route("/alembic_version", methods=["GET"])
+def alembic_version():
+    version = get_alembic_version()
+    return success_response(version)
+
+
 # Health check for server, db, and redis. Consumes latest block data from redis instead of chain.
 # Optional boolean "verbose" flag to output db connection info.
 # Optional boolean "enforce_block_diff" flag to error on unhealthy blockdiff.