Fix unique constraint add/remove in quota source labels...

galaxyproject · Feb 21, 2023 · d1a2eab · d1a2eab
1 parent f7a4992
commit d1a2eab
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/lib/galaxy/model/migrations/alembic/versions_gxy/d0583094c8cd_add_quota_source_labels.py b/lib/galaxy/model/migrations/alembic/versions_gxy/d0583094c8cd_add_quota_source_labels.py
@@ -17,7 +17,6 @@
 from galaxy.model.migrations.util import (
     add_unique_constraint,
     drop_column,
-    drop_unique_constraint,
 )
 
 # revision identifiers, used by Alembic.
@@ -39,12 +38,12 @@ def upgrade():
         Column("disk_usage", Numeric(15, 0)),
     )
     add_unique_constraint("uqsu_unique_label_per_user", "user_quota_source_usage", ["user_id", "quota_source_label"])
-    drop_unique_constraint("ix_default_quota_association_type", "default_quota_association")
+    op.drop_index("ix_default_quota_association_type", "default_quota_association")
     op.create_index("ix_quota_quota_source_label", "quota", ["quota_source_label"])
 
 
 def downgrade():
-    add_unique_constraint("ix_default_quota_association_type", "default_quota_association", ["type"])
+    op.create_index("ix_default_quota_association_type", "default_quota_association", ["type"], unique=True)
     op.drop_table("user_quota_source_usage")
     op.drop_index("ix_quota_quota_source_label", "quota")
     drop_column("quota", "quota_source_label")