Fix review issues: idempotent migration, toggle guard, picker states

This commit is contained in:
CREDO23 2026-03-31 19:27:32 +02:00
parent aecb58e22b
commit 950be8e17d
3 changed files with 51 additions and 22 deletions

View file

@ -6,6 +6,7 @@ Revises: 111
from collections.abc import Sequence
import sqlalchemy as sa
from alembic import op
revision: str = "112"
@ -31,10 +32,17 @@ def upgrade() -> None:
"CREATE INDEX IF NOT EXISTS ix_prompts_default_prompt_slug"
" ON prompts (default_prompt_slug)"
)
op.execute(
"ALTER TABLE prompts ADD CONSTRAINT uq_prompt_user_default_slug"
" UNIQUE (user_id, default_prompt_slug)"
)
conn = op.get_bind()
exists = conn.execute(
sa.text(
"SELECT 1 FROM pg_constraint WHERE conname = 'uq_prompt_user_default_slug'"
)
).scalar()
if not exists:
op.execute(
"ALTER TABLE prompts ADD CONSTRAINT uq_prompt_user_default_slug"
" UNIQUE (user_id, default_prompt_slug)"
)
op.execute(
"ALTER TABLE prompts ADD COLUMN IF NOT EXISTS"
" version INTEGER NOT NULL DEFAULT 1"