65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
|
"""补齐 0001 create_all 未覆盖的老库列（显式清单，禁止内省全库）
|
|||
|
|
|
|||
|
|
0001 对「已存在的表」不会 ALTER。下列列在 ORM / 0001 注释中视为新库默认字段，
|
|||
|
|
但 staging/production 等自 squash 前就存在的库需要本迁移显式 add_column。
|
|||
|
|
|
|||
|
|
新增 ORM 字段时：禁止改已部署 revision id；在本文件追加列定义，或新建 0020_* 迁移。
|
|||
|
|
|
|||
|
|
Revision ID: 0019_align_legacy_schema
|
|||
|
|
Revises: 0018_users_language_preference
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import sys
|
|||
|
|
from pathlib import Path
|
|||
|
|
from typing import Sequence, Union
|
|||
|
|
|
|||
|
|
import sqlalchemy as sa
|
|||
|
|
|
|||
|
|
# versions/ -> alembic/
|
|||
|
|
_ALEMBIC_DIR = Path(__file__).resolve().parents[1]
|
|||
|
|
if str(_ALEMBIC_DIR) not in sys.path:
|
|||
|
|
sys.path.insert(0, str(_ALEMBIC_DIR))
|
|||
|
|
|
|||
|
|
from script_helpers import add_column_if_missing, drop_column_if_exists, table_exists
|
|||
|
|
|
|||
|
|
revision: str = "0019_align_legacy_schema"
|
|||
|
|
down_revision: Union[str, None] = "0018_users_language_preference"
|
|||
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|||
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|||
|
|
|
|||
|
|
# (table, column) — 仅列「0002–0018 未 add_column 且老库常缺」的字段
|
|||
|
|
_LEGACY_COLUMNS: tuple[tuple[str, sa.Column], ...] = (
|
|||
|
|
(
|
|||
|
|
"segments",
|
|||
|
|
sa.Column("audio_duration_seconds", sa.Integer(), nullable=True),
|
|||
|
|
),
|
|||
|
|
(
|
|||
|
|
"conversations",
|
|||
|
|
sa.Column("deleted_at", sa.DateTime(timezone=True), nullable=True),
|
|||
|
|
),
|
|||
|
|
(
|
|||
|
|
"segments",
|
|||
|
|
sa.Column("tts_audio_urls", sa.JSON(), nullable=True),
|
|||
|
|
),
|
|||
|
|
(
|
|||
|
|
"conversation_messages",
|
|||
|
|
sa.Column("tts_audio_urls", sa.JSON(), nullable=True),
|
|||
|
|
),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def upgrade() -> None:
|
|||
|
|
for table_name, column in _LEGACY_COLUMNS:
|
|||
|
|
if not table_exists(table_name):
|
|||
|
|
continue
|
|||
|
|
add_column_if_missing(table_name, column)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def downgrade() -> None:
|
|||
|
|
for table_name, column in reversed(_LEGACY_COLUMNS):
|
|||
|
|
if not table_exists(table_name):
|
|||
|
|
continue
|
|||
|
|
drop_column_if_exists(table_name, column.name)
|