Files
ai-video-fullstack/backend/migrations/versions/20260810_0017_remove_test_suite_assistant.py
2026-08-10 13:49:24 +08:00

44 lines
1.0 KiB
Python

"""remove obsolete assistant binding from test suites
Revision ID: 20260810_0017
Revises: 20260810_0016
"""
from __future__ import annotations
from collections.abc import Sequence
from alembic import op
import sqlalchemy as sa
revision: str = "20260810_0017"
down_revision: str | Sequence[str] | None = "20260810_0016"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None
def upgrade() -> None:
op.drop_index("ix_test_suites_assistant_id", table_name="test_suites")
op.drop_column("test_suites", "assistant_id")
def downgrade() -> None:
op.add_column(
"test_suites",
sa.Column("assistant_id", sa.String(length=40), nullable=True),
)
op.create_foreign_key(
"fk_test_suites_assistant_id_assistants",
"test_suites",
"assistants",
["assistant_id"],
["id"],
ondelete="SET NULL",
)
op.create_index(
"ix_test_suites_assistant_id",
"test_suites",
["assistant_id"],
)