language-learning-app/api/alembic/versions/20260327_0005_add_translated_articles.py

44 lines
1.7 KiB
Python
Raw Permalink Normal View History

2026-03-27 11:04:05 +00:00
"""add translated_articles table
Revision ID: 0005
Revises: 0004
Create Date: 2026-03-27
"""
from typing import Sequence, Union
import sqlalchemy as sa
from alembic import op
from sqlalchemy.dialects import postgresql
revision: str = "0005"
down_revision: Union[str, None] = "0004"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
op.create_table(
"translated_articles",
sa.Column("id", postgresql.UUID(as_uuid=True), primary_key=True),
sa.Column("published_at", sa.DateTime(timezone=True), nullable=False),
sa.Column("source_language", sa.String(10), nullable=False),
sa.Column("source_title", sa.Text(), nullable=False),
sa.Column("source_body", sa.Text(), nullable=False),
sa.Column("target_language", sa.String(10), nullable=False),
sa.Column("target_complexities", postgresql.ARRAY(sa.String(5)), nullable=False),
sa.Column("target_title", sa.Text(), nullable=False),
sa.Column("target_body", sa.Text(), nullable=False),
sa.Column("audio_url", sa.Text(), nullable=True),
sa.Column("target_body_pos", postgresql.JSONB(), nullable=True),
sa.Column("target_body_transcript", postgresql.JSONB(), nullable=True),
)
op.create_index("ix_translated_articles_published_at", "translated_articles", ["published_at"])
op.create_index("ix_translated_articles_target_language", "translated_articles", ["target_language"])
def downgrade() -> None:
op.drop_index("ix_translated_articles_target_language", table_name="translated_articles")
op.drop_index("ix_translated_articles_published_at", table_name="translated_articles")
op.drop_table("translated_articles")