fix: RLS new db migration downgrade fails on SQLite (#20449)

apache · Jun 21, 2022 · ba4ba02 · ba4ba02
1 parent e3e37cb
commit ba4ba02
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/superset/migrations/versions/2022-06-19_16-17_f3afaf1f11f0_add_unique_name_desc_rls.py b/superset/migrations/versions/2022-06-19_16-17_f3afaf1f11f0_add_unique_name_desc_rls.py
@@ -72,8 +72,7 @@ def upgrade():
 
 
 def downgrade():
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.drop_constraint("uq_rls_name", "row_level_security_filters", type_="unique")
-    op.drop_column("row_level_security_filters", "description")
-    op.drop_column("row_level_security_filters", "name")
-    # ### end Alembic commands ###
+    with op.batch_alter_table("row_level_security_filters") as batch_op:
+        batch_op.drop_constraint("uq_rls_name", type_="unique")
+        batch_op.drop_column("description")
+        batch_op.drop_column("name")