apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-06-10 20:35:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

refactor: update video presentation status enum creation to use SQL execution for better handling of duplicates

Browse source
This commit is contained in:
DESKTOP-RTLN3BA\$punk 2026-03-22 01:41:15 -07:00
parent 23386b0a91
commit 0cd596b91f
1 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

15
surfsense_backend/alembic/versions/107_add_video_presentations_table.py
View file

@ -7,7 +7,7 @@ Revises: 106
from collections.abc import Sequence
import sqlalchemy as sa
from sqlalchemy.dialects.postgresql import JSONB
from sqlalchemy.dialects.postgresql import ENUM, JSONB
from alembic import op
@ -16,17 +16,24 @@ down_revision: str | None = "106"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None
video_presentation_status_enum = sa.Enum(
video_presentation_status_enum = ENUM(
"pending",
"generating",
"ready",
"failed",
name="video_presentation_status",
create_type=False,
)
def upgrade() -> None:
video_presentation_status_enum.create(op.get_bind(), checkfirst=True)
op.execute("""
DO $$ BEGIN
CREATE TYPE video_presentation_status AS ENUM ('pending', 'generating', 'ready', 'failed');
EXCEPTION
WHEN duplicate_object THEN null;
END $$;
""")
op.create_table(
"video_presentations",
@ -82,4 +89,4 @@ def downgrade() -> None:
op.drop_index("ix_video_presentations_thread_id", table_name="video_presentations")
op.drop_index("ix_video_presentations_status", table_name="video_presentations")
op.drop_table("video_presentations")
video_presentation_status_enum.drop(op.get_bind(), checkfirst=True)
op.execute("DROP TYPE IF EXISTS video_presentation_status")