mirror of
https://github.com/Monadical-SAS/reflector.git
synced 2025-12-20 20:29:06 +00:00
81 lines
2.3 KiB
Python
81 lines
2.3 KiB
Python
"""Migration transcript to text field in transcripts table
|
|
|
|
Revision ID: 9920ecfe2735
|
|
Revises: 99365b0cd87b
|
|
Create Date: 2023-11-02 18:55:17.019498
|
|
|
|
"""
|
|
|
|
from typing import Sequence, Union
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
from sqlalchemy import select
|
|
from sqlalchemy.sql import column, table
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = "9920ecfe2735"
|
|
down_revision: Union[str, None] = "99365b0cd87b"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# bind the engine
|
|
bind = op.get_bind()
|
|
|
|
# Reflect the table
|
|
transcript = table("transcript", column("id", sa.String), column("topics", sa.JSON))
|
|
|
|
# Select all rows from the transcript table
|
|
results = bind.execute(select([transcript.c.id, transcript.c.topics]))
|
|
|
|
for row in results:
|
|
transcript_id = row["id"]
|
|
topics_json = row["topics"]
|
|
|
|
# Process each topic in the topics JSON array
|
|
updated_topics = []
|
|
for topic in topics_json:
|
|
if "transcript" in topic:
|
|
# Rename key 'transcript' to 'text'
|
|
topic["text"] = topic.pop("transcript")
|
|
updated_topics.append(topic)
|
|
|
|
# Update the transcript table
|
|
bind.execute(
|
|
transcript.update()
|
|
.where(transcript.c.id == transcript_id)
|
|
.values(topics=updated_topics)
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
# bind the engine
|
|
bind = op.get_bind()
|
|
|
|
# Reflect the table
|
|
transcript = table("transcript", column("id", sa.String), column("topics", sa.JSON))
|
|
|
|
# Select all rows from the transcript table
|
|
results = bind.execute(select([transcript.c.id, transcript.c.topics]))
|
|
|
|
for row in results:
|
|
transcript_id = row["id"]
|
|
topics_json = row["topics"]
|
|
|
|
# Process each topic in the topics JSON array
|
|
updated_topics = []
|
|
for topic in topics_json:
|
|
if "text" in topic:
|
|
# Rename key 'text' back to 'transcript'
|
|
topic["transcript"] = topic.pop("text")
|
|
updated_topics.append(topic)
|
|
|
|
# Update the transcript table
|
|
bind.execute(
|
|
transcript.update()
|
|
.where(transcript.c.id == transcript_id)
|
|
.values(topics=updated_topics)
|
|
)
|