Inject table for fixtures processing in all binds

Author: febus982

src/alembic/env.py

@@ -1,7 +1,10 @@
 import logging
 from asyncio import get_event_loop
+from datetime import datetime
 
+from sqlalchemy import Table, Column, String, DateTime
 from sqlalchemy.ext.asyncio import AsyncEngine
+from sqlalchemy.orm import registry
 
 from alembic import context
 from common.bootstrap import application_init
@@ -32,6 +35,19 @@
 db_names = target_metadata.keys()
 config.set_main_option("databases", ",".join(db_names))
 
+def inject_fixture_tables(registry_mapper: registry):
+    Table(
+        "alembic_fixtures",
+        registry_mapper.metadata,
+        Column("filename", String(), primary_key=True),
+        Column("signature", String(), nullable=False),
+        Column("processed_at", DateTime(timezone=True), nullable=False, default=datetime.now),
+    )
+
+for name in db_names:
+    inject_fixture_tables(sa_manager.get_bind(name).registry_mapper)
+
+
 # add your model's MetaData objects here
 # for 'autogenerate' support.  These must be set
 # up to hold just those tables targeting a

src/alembic/versions/2025-01-26-212326-52b1246eda46_initialize_fixture_tables.py

@@ -0,0 +1,38 @@
+"""Initialize fixture tables
+
+Revision ID: 52b1246eda46
+Revises:
+Create Date: 2025-01-26 21:23:26.321986
+
+"""
+
+import sqlalchemy as sa
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "52b1246eda46"
+down_revision = None
+branch_labels = None
+depends_on = None
+
+
+def upgrade(engine_name: str) -> None:
+    globals()[f"upgrade_{engine_name}"]()
+
+
+def downgrade(engine_name: str) -> None:
+    globals()[f"downgrade_{engine_name}"]()
+
+
+def upgrade_default() -> None:
+    op.create_table('alembic_fixtures',
+        sa.Column('filename', sa.String(), nullable=False),
+        sa.Column('signature', sa.String(), nullable=False),
+        sa.Column('processed_at', sa.DateTime(timezone=True), nullable=False),
+        sa.PrimaryKeyConstraint('filename')
+    )
+
+
+def downgrade_default() -> None:
+    op.drop_table('alembic_fixtures')

src/alembic/versions/2022-11-09-203313-52b1246eda46_create_tables.py renamed to src/alembic/versions/2025-01-26-212826-bd73bd8a2ac4_create_books_table.py

@@ -1,18 +1,17 @@
-"""create books table
+"""Create books table
 
-Revision ID: 52b1246eda46
-Revises:
-Create Date: 2022-11-09 20:33:13.035514
+Revision ID: bd73bd8a2ac4
+Revises: 52b1246eda46
+Create Date: 2025-01-26 21:28:26.321986
 
 """
-
+from alembic import op
 import sqlalchemy as sa
 
-from alembic import op
 
 # revision identifiers, used by Alembic.
-revision = "52b1246eda46"
-down_revision = None
+revision = 'bd73bd8a2ac4'
+down_revision = '52b1246eda46'
 branch_labels = None
 depends_on = None