Stamp alembic head when recreating tables

backend/bracket/app.py

@@ -9,8 +9,6 @@
 from starlette.responses import JSONResponse, Response
 from starlette.staticfiles import StaticFiles
 
-from alembic import command
-from alembic.config import Config
 from bracket.config import Environment, config, environment, init_sentry
 from bracket.cronjobs.scheduling import start_cronjobs
 from bracket.database import database
@@ -29,26 +27,21 @@
     tournaments,
     users,
 )
+from bracket.utils.alembic import alembic_run_migrations
 from bracket.utils.asyncio import AsyncioTasksManager
 from bracket.utils.db_init import init_db_when_empty
 from bracket.utils.logging import logger
 
 init_sentry()
 
 
-def run_migrations() -> None:
-    logger.info("Running migrations")
-    alembic_cfg = Config("alembic.ini")
-    command.upgrade(alembic_cfg, "head")
-
-
 @asynccontextmanager
 async def lifespan(_: FastAPI) -> AsyncIterator[None]:
     await database.connect()
     await init_db_when_empty()
 
     if config.auto_run_migrations and environment is not Environment.CI:
-        run_migrations()
+        alembic_run_migrations()
 
     if environment is Environment.PRODUCTION:
         start_cronjobs()

backend/bracket/utils/alembic.py

@@ -0,0 +1,17 @@
+from alembic import command
+from alembic.config import Config
+from bracket.utils.logging import logger
+
+
+def get_alembic_config() -> Config:
+    return Config("alembic.ini")
+
+
+def alembic_run_migrations() -> None:
+    logger.info("Running migrations")
+    command.upgrade(get_alembic_config(), "head")
+
+
+def alembic_stamp_head() -> None:
+    logger.info("Overwriting current version to be the latest revision (head)")
+    command.stamp(get_alembic_config(), "head")

backend/bracket/utils/db_init.py

@@ -44,6 +44,7 @@
 from bracket.sql.stages import get_full_tournament_details
 from bracket.sql.tournaments import sql_get_tournament
 from bracket.sql.users import create_user, get_user
+from bracket.utils.alembic import alembic_stamp_head
 from bracket.utils.db import insert_generic
 from bracket.utils.dummy_records import (
     DUMMY_CLUB,
@@ -116,6 +117,7 @@
         ):
             logger.warning("Empty db detected, creating tables...")
             metadata.create_all(engine)
+            alembic_stamp_head()
 
             logger.warning("Empty db detected, creating admin user...")
             return await create_admin_user()
@@ -133,6 +135,7 @@
     metadata.drop_all(engine)
     metadata.create_all(engine)
     real_user_id = await init_db_when_empty()
+    alembic_stamp_head()
 
     table_lookup: dict[type, Table] = {
         User: users,