40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
"""segments: rename transcript_text -> user_input_text
|
|
|
|
语义：用户输入正文（语音 ASR 或键盘输入），不仅限于「转写」。
|
|
|
|
Revision ID: 0002_segments_user_input_text
|
|
Revises: 0001_initial
|
|
"""
|
|
|
|
from typing import Sequence, Union
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
revision: str = "0002_segments_user_input_text"
|
|
down_revision: Union[str, None] = "0001_initial"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def _column_names(table_name: str) -> set[str]:
|
|
bind = op.get_bind()
|
|
inspector = sa.inspect(bind)
|
|
return {column["name"] for column in inspector.get_columns(table_name)}
|
|
|
|
|
|
def upgrade() -> None:
|
|
columns = _column_names("segments")
|
|
if "transcript_text" in columns and "user_input_text" not in columns:
|
|
op.execute(
|
|
"ALTER TABLE segments RENAME COLUMN transcript_text TO user_input_text"
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
columns = _column_names("segments")
|
|
if "user_input_text" in columns and "transcript_text" not in columns:
|
|
op.execute(
|
|
"ALTER TABLE segments RENAME COLUMN user_input_text TO transcript_text"
|
|
)
|