diff --git a/src/ai/backend/manager/models/alembic/versions/2be0ce116c35_add_vfolder_permissions_id.py b/src/ai/backend/manager/models/alembic/versions/2be0ce116c35_add_vfolder_permissions_id.py deleted file mode 100644 index f6f67840172..00000000000 --- a/src/ai/backend/manager/models/alembic/versions/2be0ce116c35_add_vfolder_permissions_id.py +++ /dev/null @@ -1,31 +0,0 @@ -"""add_id_to_vfolder_permissions - -Revision ID: 2be0ce116c35 -Revises: 37410c773b8c -Create Date: 2024-05-11 00:53:28.387238 - -""" - -import sqlalchemy as sa -from alembic import op - -from ai.backend.manager.models.base import GUID - -# revision identifiers, used by Alembic. -revision = "2be0ce116c35" -down_revision = "37410c773b8c" -branch_labels = None -depends_on = None - - -def upgrade(): - op.add_column( - "vfolder_permissions", - sa.Column("id", GUID(), server_default=sa.text("uuid_generate_v4()"), nullable=False), - ) - op.create_primary_key("pk_vfolder_permissions", "vfolder_permissions", ["id"]) - - -def downgrade(): - op.drop_constraint("pk_vfolder_permissions", "vfolder_permissions", type_="primary") - op.drop_column("vfolder_permissions", "id") diff --git a/src/ai/backend/manager/models/alembic/versions/37410c773b8c_add_vfolders_domain_name.py b/src/ai/backend/manager/models/alembic/versions/37410c773b8c_add_vfolders_domain_name.py deleted file mode 100644 index c2a4802e2ac..00000000000 --- a/src/ai/backend/manager/models/alembic/versions/37410c773b8c_add_vfolders_domain_name.py +++ /dev/null @@ -1,43 +0,0 @@ -"""add_vfolders_domain_name - -Revision ID: 37410c773b8c -Revises: f56a82d0ac9f -Create Date: 2024-05-06 20:51:54.658829 - -""" - -import sqlalchemy as sa -from alembic import op -from sqlalchemy.sql import text - -# revision identifiers, used by Alembic. -revision = "37410c773b8c" -down_revision = "f56a82d0ac9f" -branch_labels = None -depends_on = None - - -def upgrade(): - conn = op.get_bind() - op.add_column("vfolders", sa.Column("domain_name", sa.String(length=64), nullable=True)) - - conn.execute( - text( - """\ - UPDATE vfolders - SET domain_name = COALESCE( - (SELECT domain_name FROM users WHERE vfolders.user = users.uuid), - (SELECT domain_name FROM groups WHERE vfolders.group = groups.id) - ) - WHERE domain_name IS NULL; - """ - ) - ) - - op.alter_column("vfolders", column_name="domain_name", nullable=False) - op.create_index(op.f("ix_vfolders_domain_name"), "vfolders", ["domain_name"], unique=False) - - -def downgrade(): - op.drop_index(op.f("ix_vfolders_domain_name"), table_name="vfolders") - op.drop_column("vfolders", "domain_name")