1ae32ffc8c3f_rename_data_source_unique_constraint.py 847 B

123456789101112131415161718192021222324252627282930313233343536
  1. """Rename data source unique constraint
  2. Revision ID: 1ae32ffc8c3f
  3. Revises: 30fe2267e7d5
  4. Create Date: 2021-11-11 16:54:09.302274
  5. """
  6. from alembic import op
  7. # revision identifiers, used by Alembic.
  8. revision = "1ae32ffc8c3f"
  9. down_revision = "30fe2267e7d5"
  10. branch_labels = None
  11. depends_on = None
  12. def upgrade():
  13. op.drop_constraint(
  14. "_data_source_name_user_id_model_version_key", "data_source", type_="unique"
  15. )
  16. op.create_unique_constraint(
  17. op.f("data_source_name_key"),
  18. "data_source",
  19. ["name", "user_id", "model", "version"],
  20. )
  21. def downgrade():
  22. op.drop_constraint(op.f("data_source_name_key"), "data_source", type_="unique")
  23. op.create_unique_constraint(
  24. "_data_source_name_user_id_model_version_key",
  25. "data_source",
  26. ["name", "user_id", "model", "version"],
  27. )