xpltdco/chrysopedia
2
0
Fork 0
Code Issues 2 Pull requests Projects Releases Packages Wiki Activity Actions

fix: migration 008 — use text column conversion to avoid PG enum transaction restriction

Browse source
This commit is contained in:
jlightner 2026-03-31 02:45:54 +00:00
parent 35fc699098
commit 63225a3559
1 changed files with 47 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

54
alembic/versions/008_rename_processing_status.py
View file

@ -3,10 +3,13 @@
Old: pending, transcribed, extracted, published Old: pending, transcribed, extracted, published
New: not_started, queued, processing, error, complete New: not_started, queued, processing, error, complete
Uses text column conversion to avoid PG enum ADD VALUE transaction restriction.
Revision ID: 008_rename_processing_status Revision ID: 008_rename_processing_status
Revises: 007_drop_review_columns Revises: 007_drop_review_columns
""" """
from alembic import op from alembic import op
import sqlalchemy as sa
revision = "008_rename_processing_status" revision = "008_rename_processing_status"
down_revision = "007_drop_review_columns" down_revision = "007_drop_review_columns"
@ -15,21 +18,58 @@ depends_on = None
def upgrade() -> None: def upgrade() -> None:
# Add new enum values first # 1. Convert column to text to break free of the old enum
op.execute("ALTER TYPE processing_status ADD VALUE IF NOT EXISTS 'not_started'") op.alter_column(
op.execute("ALTER TYPE processing_status ADD VALUE IF NOT EXISTS 'queued'") "source_videos", "processing_status",
op.execute("ALTER TYPE processing_status ADD VALUE IF NOT EXISTS 'processing'") type_=sa.Text(),
op.execute("ALTER TYPE processing_status ADD VALUE IF NOT EXISTS 'error'") existing_type=sa.Enum(name="processing_status"),
op.execute("ALTER TYPE processing_status ADD VALUE IF NOT EXISTS 'complete'") postgresql_using="processing_status::text",
# Migrate existing rows )
# 2. Drop old enum type
op.execute("DROP TYPE IF EXISTS processing_status")
# 3. Rename values in the text column
op.execute("UPDATE source_videos SET processing_status = 'not_started' WHERE processing_status = 'pending'") op.execute("UPDATE source_videos SET processing_status = 'not_started' WHERE processing_status = 'pending'")
op.execute("UPDATE source_videos SET processing_status = 'queued' WHERE processing_status = 'transcribed'") op.execute("UPDATE source_videos SET processing_status = 'queued' WHERE processing_status = 'transcribed'")
op.execute("UPDATE source_videos SET processing_status = 'processing' WHERE processing_status = 'extracted'") op.execute("UPDATE source_videos SET processing_status = 'processing' WHERE processing_status = 'extracted'")
op.execute("UPDATE source_videos SET processing_status = 'complete' WHERE processing_status = 'published'") op.execute("UPDATE source_videos SET processing_status = 'complete' WHERE processing_status = 'published'")
# 4. Create new enum type
processing_status = sa.Enum(
"not_started", "queued", "processing", "error", "complete",
name="processing_status",
)
processing_status.create(op.get_bind(), checkfirst=True)
# 5. Convert column back to enum
op.alter_column(
"source_videos", "processing_status",
type_=processing_status,
existing_type=sa.Text(),
postgresql_using="processing_status::processing_status",
server_default="not_started",
)
def downgrade() -> None: def downgrade() -> None:
op.alter_column(
"source_videos", "processing_status",
type_=sa.Text(),
existing_type=sa.Enum(name="processing_status"),
postgresql_using="processing_status::text",
)
op.execute("DROP TYPE IF EXISTS processing_status")
op.execute("UPDATE source_videos SET processing_status = 'pending' WHERE processing_status = 'not_started'") op.execute("UPDATE source_videos SET processing_status = 'pending' WHERE processing_status = 'not_started'")
op.execute("UPDATE source_videos SET processing_status = 'transcribed' WHERE processing_status = 'queued'") op.execute("UPDATE source_videos SET processing_status = 'transcribed' WHERE processing_status = 'queued'")
op.execute("UPDATE source_videos SET processing_status = 'extracted' WHERE processing_status = 'processing'") op.execute("UPDATE source_videos SET processing_status = 'extracted' WHERE processing_status = 'processing'")
op.execute("UPDATE source_videos SET processing_status = 'published' WHERE processing_status = 'complete'") op.execute("UPDATE source_videos SET processing_status = 'published' WHERE processing_status = 'complete'")
old_enum = sa.Enum("pending", "transcribed", "extracted", "published", name="processing_status")
old_enum.create(op.get_bind(), checkfirst=True)
op.alter_column(
"source_videos", "processing_status",
type_=old_enum,
existing_type=sa.Text(),
postgresql_using="processing_status::processing_status",
server_default="pending",
)