coulomb/hub-core
4
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: commit migration session setup
Some checks failed
CI Smoke / host-smoke (push) Successful in 0s
Details
CI Smoke / pytest-smoke (push) Failing after 2s
Details

Browse source 
Assistant: codex
Assistant-Model: gpt-5.6-sol
Assistant-Session: 01a053ff-1d6f-7fe2-ac1c-a6eb40a42a0c
This commit is contained in:
tegwick 2026-09-01 02:05:56 +02:00
parent 2dcec7718c
commit dda93db0ac
3 changed files with 68 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

11
hub_core/migrations/env.py
View file

@ -4,8 +4,8 @@ from logging.config import fileConfig
from alembic import context
from sqlalchemy import engine_from_config, pool
from hub_core.migrations.roles import configure_migration_session
from hub_core.models import Base
from hub_core.migrations.roles import migration_role_statement, migration_schema_statement
config = context.config
@ -39,10 +39,11 @@ def run_migrations_online() -> None:
poolclass=pool.NullPool,
)
with connectable.connect() as connection:
if statement := migration_role_statement(os.environ.get("HUB_CORE_MIGRATION_ROLE")):
connection.exec_driver_sql(statement)
if statement := migration_schema_statement(os.environ.get("HUB_CORE_MIGRATION_SCHEMA")):
connection.exec_driver_sql(statement)
configure_migration_session(
connection,
role=os.environ.get("HUB_CORE_MIGRATION_ROLE"),
schema=os.environ.get("HUB_CORE_MIGRATION_SCHEMA"),
)
context.configure(connection=connection, target_metadata=target_metadata)
with context.begin_transaction():
context.run_migrations()