Merge "db: Don't rely on branched connections"

2024-04-15 17:09:44 +00:00 · 2024-04-15 17:09:44 +00:00 · 3ca2d4fdc8
commit 3ca2d4fdc8
parent 6dba127ad0 d683ddb86e
1 changed files with 9 additions and 4 deletions
--- a/taskflow/persistence/backends/sqlalchemy/alembic/env.py
+++ b/taskflow/persistence/backends/sqlalchemy/alembic/env.py
@ -68,7 +68,12 @@ def run_migrations_online():
                              target_metadata=target_metadata)
            with context.begin_transaction():
                context.run_migrations()
-
+    else:
+        context.configure(
+            connection=connectable,
+            target_metadata=target_metadata)
+        with context.begin_transaction():
+            context.run_migrations()

 if context.is_offline_mode():
    run_migrations_offline()