copilot-swe-agent[bot]
77df9628c3
Co-authored-by: christianlouis <361235+christianlouis@users.noreply.github.com>
71 lines
3.2 KiB
Python
71 lines
3.2 KiB
Python
"""Add document translation fields to files and user_profiles tables.
|
|
|
|
Adds detected_language, default_language_text, and default_language_code to
|
|
the files table so that a translated version of the document text can be
|
|
stored alongside the original.
|
|
|
|
Adds default_document_language to user_profiles so each user can override
|
|
the system-wide default translation target language.
|
|
|
|
Revision ID: 036_add_document_translation_fields
|
|
Revises: 035_add_routing_rules
|
|
Create Date: 2026-03-16
|
|
"""
|
|
|
|
from typing import Union
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
revision: str = "036_add_document_translation_fields"
|
|
down_revision: Union[str, None] = "035_add_routing_rules"
|
|
depends_on: Union[str, None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
"""Add translation columns to files and user_profiles."""
|
|
conn = op.get_bind()
|
|
inspector = sa.inspect(conn)
|
|
|
|
if "files" in inspector.get_table_names():
|
|
existing_file_cols = {col["name"] for col in inspector.get_columns("files")}
|
|
cols_to_add = {"detected_language", "default_language_text", "default_language_code"} - existing_file_cols
|
|
if cols_to_add:
|
|
with op.batch_alter_table("files") as batch_op:
|
|
if "detected_language" in cols_to_add:
|
|
batch_op.add_column(sa.Column("detected_language", sa.String(10), nullable=True))
|
|
if "default_language_text" in cols_to_add:
|
|
batch_op.add_column(sa.Column("default_language_text", sa.Text(), nullable=True))
|
|
if "default_language_code" in cols_to_add:
|
|
batch_op.add_column(sa.Column("default_language_code", sa.String(10), nullable=True))
|
|
|
|
if "user_profiles" in inspector.get_table_names():
|
|
existing_profile_cols = {col["name"] for col in inspector.get_columns("user_profiles")}
|
|
if "default_document_language" not in existing_profile_cols:
|
|
with op.batch_alter_table("user_profiles") as batch_op:
|
|
batch_op.add_column(sa.Column("default_document_language", sa.String(10), nullable=True))
|
|
|
|
|
|
def downgrade() -> None:
|
|
"""Remove translation columns."""
|
|
conn = op.get_bind()
|
|
inspector = sa.inspect(conn)
|
|
|
|
if "user_profiles" in inspector.get_table_names():
|
|
existing_profile_cols = {col["name"] for col in inspector.get_columns("user_profiles")}
|
|
if "default_document_language" in existing_profile_cols:
|
|
with op.batch_alter_table("user_profiles") as batch_op:
|
|
batch_op.drop_column("default_document_language")
|
|
|
|
if "files" in inspector.get_table_names():
|
|
existing_file_cols = {col["name"] for col in inspector.get_columns("files")}
|
|
cols_to_drop = {"detected_language", "default_language_text", "default_language_code"} & existing_file_cols
|
|
if cols_to_drop:
|
|
with op.batch_alter_table("files") as batch_op:
|
|
if "default_language_code" in cols_to_drop:
|
|
batch_op.drop_column("default_language_code")
|
|
if "default_language_text" in cols_to_drop:
|
|
batch_op.drop_column("default_language_text")
|
|
if "detected_language" in cols_to_drop:
|
|
batch_op.drop_column("detected_language")
|