Wrap batch ops in pragma statement for sqlite

See sqlalchemy/alembic#1207
jdavcs · Apr 13, 2023 · 210bbef · 210bbef
1 parent cf52f76
commit 210bbef
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/lib/galaxy/model/migrations/util.py b/lib/galaxy/model/migrations/util.py
@@ -27,8 +27,10 @@ def add_column(table_name: str, column: sa.Column) -> None:
         log.info("Generation of `alter` statements is disabled in offline mode.")
         return
     if _is_sqlite():
+        op.execute("PRAGMA legacy_alter_table=1;")
         with op.batch_alter_table(table_name) as batch_op:
             batch_op.add_column(column)
+        op.execute("PRAGMA legacy_alter_table=0;")
     else:
         op.add_column(table_name, column)
 
@@ -38,8 +40,10 @@ def drop_column(table_name, column_name):
         log.info("Generation of `alter` statements is disabled in offline mode.")
         return
     if _is_sqlite():
+        op.execute("PRAGMA legacy_alter_table=1;")
         with op.batch_alter_table(table_name) as batch_op:
             batch_op.drop_column(column_name)
+        op.execute("PRAGMA legacy_alter_table=0;")
     else:
         op.drop_column(table_name, column_name)