update migrations to handle sqllite

superset/migrations/versions/2024-05-24_11-31_02f4f7811799_remove_sl_dataset_columns_table.py

@@ -33,8 +33,18 @@
 
 
 def upgrade():
-    drop_fks_for_table("sl_dataset_columns")
-    op.drop_table("sl_dataset_columns")
+    bind = op.get_bind()
+
+    # Determine if we're using SQLite
+    if bind.dialect.name == "sqlite":
+        # Use batch mode for SQLite
+        with op.batch_alter_table("sl_dataset_columns") as batch_op:
+            drop_fks_for_table("sl_dataset_columns")
+            batch_op.drop_table("sl_dataset_columns")
+    else:
+        # Non-SQLite: Directly drop the foreign keys and table
+        drop_fks_for_table("sl_dataset_columns")
+        op.drop_table("sl_dataset_columns")
 
 
 def downgrade():

superset/migrations/versions/2024-08-13_15-17_39549add7bfc_remove_sl_table_columns_table.py

@@ -33,8 +33,18 @@
 
 
 def upgrade():
-    drop_fks_for_table("sl_table_columns")
-    op.drop_table("sl_table_columns")
+    bind = op.get_bind()
+
+    # Determine if we're using SQLite
+    if bind.dialect.name == "sqlite":
+        # Use batch mode for SQLite
+        with op.batch_alter_table("sl_table_columns") as batch_op:
+            drop_fks_for_table("sl_table_columns")
+            batch_op.drop_table("sl_table_columns")
+    else:
+        # Non-SQLite: Directly drop the foreign keys and table
+        drop_fks_for_table("sl_table_columns")
+        op.drop_table("sl_table_columns")
 
 
 def downgrade():

superset/migrations/versions/2024-08-13_15-23_38f4144e8558_remove_sl_dataset_tables.py

@@ -33,8 +33,18 @@
 
 
 def upgrade():
-    drop_fks_for_table("sl_dataset_tables")
-    op.drop_table("sl_dataset_tables")
+    bind = op.get_bind()
+
+    # Determine if we're using SQLite
+    if bind.dialect.name == "sqlite":
+        # Use batch mode for SQLite
+        with op.batch_alter_table("sl_dataset_tables") as batch_op:
+            drop_fks_for_table("sl_dataset_tables")
+            batch_op.drop_table("sl_dataset_tables")
+    else:
+        # Non-SQLite: Directly drop the foreign keys and table
+        drop_fks_for_table("sl_dataset_tables")
+        op.drop_table("sl_dataset_tables")
 
 
 def downgrade():

superset/migrations/versions/2024-08-13_15-27_e53fd48cc078_remove_sl_dataset_users.py

@@ -33,8 +33,18 @@
 
 
 def upgrade():
-    drop_fks_for_table("sl_dataset_users")
-    op.drop_table("sl_dataset_users")
+    bind = op.get_bind()
+
+    # Determine if we're using SQLite
+    if bind.dialect.name == "sqlite":
+        # Use batch mode for SQLite
+        with op.batch_alter_table("sl_dataset_users") as batch_op:
+            drop_fks_for_table("sl_dataset_users")
+            batch_op.drop_table("sl_dataset_users")
+    else:
+        # Non-SQLite: Directly drop the foreign keys and table
+        drop_fks_for_table("sl_dataset_users")
+        op.drop_table("sl_dataset_users")
 
 
 def downgrade():

superset/migrations/versions/2024-08-13_15-29_a6b32d2d07b1_remove_sl_columns.py

@@ -33,8 +33,18 @@
 
 
 def upgrade():
-    drop_fks_for_table("sl_columns")
-    op.drop_table("sl_columns")
+    bind = op.get_bind()
+
+    # Determine if we're using SQLite
+    if bind.dialect.name == "sqlite":
+        # Use batch mode for SQLite
+        with op.batch_alter_table("sl_columns") as batch_op:
+            drop_fks_for_table("sl_columns")
+            batch_op.drop_table("sl_columns")
+    else:
+        # Non-SQLite: Directly drop the foreign keys and table
+        drop_fks_for_table("sl_columns")
+        op.drop_table("sl_columns")
 
 
 def downgrade():

superset/migrations/versions/2024-08-13_15-31_007a1abffe7e_remove_sl_tables.py

@@ -33,8 +33,18 @@
 
 
 def upgrade():
-    drop_fks_for_table("sl_tables")
-    op.drop_table("sl_tables")
+    bind = op.get_bind()
+
+    # Determine if we're using SQLite
+    if bind.dialect.name == "sqlite":
+        # Use batch mode for SQLite
+        with op.batch_alter_table("sl_tables") as batch_op:
+            drop_fks_for_table("sl_tables")
+            batch_op.drop_table("sl_tables")
+    else:
+        # Non-SQLite: Directly drop the foreign keys and table
+        drop_fks_for_table("sl_tables")
+        op.drop_table("sl_tables")
 
 
 def downgrade():

superset/migrations/versions/2024-08-13_15-33_48cbb571fa3a_remove_sl_datasets.py

@@ -33,8 +33,18 @@
 
 
 def upgrade():
-    drop_fks_for_table("sl_datasets")
-    op.drop_table("sl_datasets")
+    bind = op.get_bind()
+
+    # Determine if we're using SQLite
+    if bind.dialect.name == "sqlite":
+        # Use batch mode for SQLite
+        with op.batch_alter_table("sl_datasets") as batch_op:
+            drop_fks_for_table("sl_datasets")
+            batch_op.drop_table("sl_datasets")
+    else:
+        # Non-SQLite: Directly drop the foreign keys and table
+        drop_fks_for_table("sl_datasets")
+        op.drop_table("sl_datasets")
 
 
 def downgrade():