119 lines
4.7 KiB
Python
119 lines
4.7 KiB
Python
"""add ai generation runs and item variant lifecycle fields
|
|
|
|
Revision ID: 20260404_000003
|
|
Revises: 20260402_000002
|
|
Create Date: 2026-04-04 10:10:00
|
|
"""
|
|
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
revision: str = "20260404_000003"
|
|
down_revision: Union[str, None] = "20260402_000002"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.create_table(
|
|
"ai_generation_runs",
|
|
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
|
|
sa.Column("basis_item_id", sa.Integer(), nullable=False),
|
|
sa.Column("source_snapshot_question_id", sa.Integer(), nullable=True),
|
|
sa.Column("target_level", sa.String(length=50), nullable=False),
|
|
sa.Column("requested_count", sa.Integer(), nullable=False, server_default="1"),
|
|
sa.Column("model", sa.String(length=255), nullable=False),
|
|
sa.Column("prompt_version", sa.String(length=50), nullable=False, server_default="v1"),
|
|
sa.Column("operator_notes", sa.Text(), nullable=True),
|
|
sa.Column("created_by", sa.String(length=255), nullable=False),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
|
|
sa.ForeignKeyConstraint(["basis_item_id"], ["items.id"], ondelete="CASCADE", onupdate="CASCADE"),
|
|
sa.ForeignKeyConstraint(
|
|
["source_snapshot_question_id"],
|
|
["tryout_snapshot_questions.id"],
|
|
ondelete="SET NULL",
|
|
onupdate="CASCADE",
|
|
),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index("ix_ai_generation_runs_basis_item_id", "ai_generation_runs", ["basis_item_id"], unique=False)
|
|
op.create_index(
|
|
"ix_ai_generation_runs_source_snapshot_question_id",
|
|
"ai_generation_runs",
|
|
["source_snapshot_question_id"],
|
|
unique=False,
|
|
)
|
|
|
|
op.add_column("items", sa.Column("generation_run_id", sa.Integer(), nullable=True))
|
|
op.add_column("items", sa.Column("source_snapshot_question_id", sa.Integer(), nullable=True))
|
|
op.add_column("items", sa.Column("variant_status", sa.String(length=50), nullable=False, server_default="active"))
|
|
op.add_column("items", sa.Column("reviewed_by", sa.String(length=255), nullable=True))
|
|
op.add_column("items", sa.Column("reviewed_at", sa.DateTime(timezone=True), nullable=True))
|
|
op.add_column("items", sa.Column("review_notes", sa.Text(), nullable=True))
|
|
|
|
op.create_foreign_key(
|
|
"fk_items_generation_run_id",
|
|
"items",
|
|
"ai_generation_runs",
|
|
["generation_run_id"],
|
|
["id"],
|
|
ondelete="SET NULL",
|
|
onupdate="CASCADE",
|
|
)
|
|
op.create_foreign_key(
|
|
"fk_items_source_snapshot_question_id",
|
|
"items",
|
|
"tryout_snapshot_questions",
|
|
["source_snapshot_question_id"],
|
|
["id"],
|
|
ondelete="SET NULL",
|
|
onupdate="CASCADE",
|
|
)
|
|
op.create_index("ix_items_generation_run_id", "items", ["generation_run_id"], unique=False)
|
|
op.create_index(
|
|
"ix_items_source_snapshot_question_id",
|
|
"items",
|
|
["source_snapshot_question_id"],
|
|
unique=False,
|
|
)
|
|
op.create_index("ix_items_variant_status", "items", ["variant_status"], unique=False)
|
|
|
|
op.drop_index("ix_items_tryout_id_website_id_slot", table_name="items")
|
|
op.create_index(
|
|
"ix_items_tryout_id_website_id_slot",
|
|
"items",
|
|
["tryout_id", "website_id", "slot", "level"],
|
|
unique=False,
|
|
)
|
|
|
|
op.alter_column("items", "variant_status", server_default=None)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index("ix_items_tryout_id_website_id_slot", table_name="items")
|
|
op.create_index(
|
|
"ix_items_tryout_id_website_id_slot",
|
|
"items",
|
|
["tryout_id", "website_id", "slot", "level"],
|
|
unique=True,
|
|
)
|
|
|
|
op.drop_index("ix_items_variant_status", table_name="items")
|
|
op.drop_index("ix_items_source_snapshot_question_id", table_name="items")
|
|
op.drop_index("ix_items_generation_run_id", table_name="items")
|
|
op.drop_constraint("fk_items_source_snapshot_question_id", "items", type_="foreignkey")
|
|
op.drop_constraint("fk_items_generation_run_id", "items", type_="foreignkey")
|
|
op.drop_column("items", "review_notes")
|
|
op.drop_column("items", "reviewed_at")
|
|
op.drop_column("items", "reviewed_by")
|
|
op.drop_column("items", "variant_status")
|
|
op.drop_column("items", "source_snapshot_question_id")
|
|
op.drop_column("items", "generation_run_id")
|
|
|
|
op.drop_index("ix_ai_generation_runs_source_snapshot_question_id", table_name="ai_generation_runs")
|
|
op.drop_index("ix_ai_generation_runs_basis_item_id", table_name="ai_generation_runs")
|
|
op.drop_table("ai_generation_runs")
|