Files
xiaoxia-saas/alembic/versions/010_phase0_extensibility.py
T
xiaoxia 1217d8cef0
Deploy / Build Production Runtime Images (push) Has been skipped
Deploy / Deploy Production (push) Has been skipped
Deploy / Production Browser E2E (push) Has been skipped
Deploy / Deploy Staging (push) Failing after 210h35m44s
CI/CD Pipeline / Frontend Lint (push) Failing after 210h36m11s
CI/CD Pipeline / Validate Code Quality And Tests (push) Failing after 210h36m17s
style: apply isort formatting to pass CI validation (#129)
2026-06-30 18:17:44 +08:00

98 lines
4.2 KiB
Python

"""Phase 0 - 扩展性基础设施:metadata JSONB + title_libraries + voice_libraries
Revision ID: 010
Revises: 009
Create Date: 2026-06-28
This migration:
1. Adds metadata JSONB column to 5 tables:
- projects, asset_libraries, assets, edit_templates, generation_tasks
2. Creates title_libraries table (独立标题库,支持跨项目复用)
3. Creates voice_libraries table (配音库,支持 AI 配音管理)
"""
import sqlalchemy as sa
from alembic import op
# revision identifiers
revision = "010"
down_revision = "009"
branch_labels = None
depends_on = None
def upgrade() -> None:
conn = op.get_bind()
# ── 1. Add metadata JSONB to existing tables ──
conn.execute(sa.text("ALTER TABLE projects ADD COLUMN IF NOT EXISTS metadata JSONB NOT NULL DEFAULT '{}'"))
conn.execute(sa.text("ALTER TABLE asset_libraries ADD COLUMN IF NOT EXISTS metadata JSONB NOT NULL DEFAULT '{}'"))
conn.execute(sa.text("ALTER TABLE assets ADD COLUMN IF NOT EXISTS metadata JSONB NOT NULL DEFAULT '{}'"))
conn.execute(sa.text("ALTER TABLE edit_templates ADD COLUMN IF NOT EXISTS metadata JSONB NOT NULL DEFAULT '{}'"))
conn.execute(sa.text("ALTER TABLE generation_tasks ADD COLUMN IF NOT EXISTS metadata JSONB NOT NULL DEFAULT '{}'"))
# ── 2. Create title_libraries table ──
conn.execute(sa.text("""
CREATE TABLE IF NOT EXISTS title_libraries (
id VARCHAR(36) PRIMARY KEY,
user_id VARCHAR(36) NOT NULL,
name VARCHAR(255) NOT NULL,
description TEXT NOT NULL DEFAULT '',
category VARCHAR(50) NOT NULL DEFAULT 'default',
text VARCHAR(500) NOT NULL,
tags JSONB NOT NULL DEFAULT '[]',
usage_count INTEGER NOT NULL DEFAULT 0,
is_active BOOLEAN NOT NULL DEFAULT TRUE,
metadata JSONB NOT NULL DEFAULT '{}',
created_at TIMESTAMP NOT NULL DEFAULT NOW(),
updated_at TIMESTAMP NOT NULL DEFAULT NOW()
)
"""))
conn.execute(sa.text("CREATE INDEX IF NOT EXISTS ix_title_libraries_user_id ON title_libraries(user_id)"))
conn.execute(sa.text("CREATE INDEX IF NOT EXISTS ix_title_libraries_category ON title_libraries(category)"))
conn.execute(sa.text("CREATE INDEX IF NOT EXISTS ix_title_libraries_is_active ON title_libraries(is_active)"))
# ── 3. Create voice_libraries table ──
conn.execute(sa.text("""
CREATE TABLE IF NOT EXISTS voice_libraries (
id VARCHAR(36) PRIMARY KEY,
user_id VARCHAR(36) NOT NULL,
project_id VARCHAR(36),
name VARCHAR(255) NOT NULL,
text TEXT NOT NULL DEFAULT '',
voice_provider VARCHAR(50) NOT NULL DEFAULT '',
voice_id VARCHAR(100) NOT NULL DEFAULT '',
voice_name VARCHAR(100) NOT NULL DEFAULT '',
audio_url VARCHAR(1000) NOT NULL DEFAULT '',
duration FLOAT NOT NULL DEFAULT 0,
file_size INTEGER NOT NULL DEFAULT 0,
status VARCHAR(20) NOT NULL DEFAULT 'completed',
tags JSONB NOT NULL DEFAULT '[]',
metadata JSONB NOT NULL DEFAULT '{}',
created_at TIMESTAMP NOT NULL DEFAULT NOW(),
updated_at TIMESTAMP NOT NULL DEFAULT NOW()
)
"""))
conn.execute(sa.text("CREATE INDEX IF NOT EXISTS ix_voice_libraries_user_id ON voice_libraries(user_id)"))
conn.execute(sa.text("CREATE INDEX IF NOT EXISTS ix_voice_libraries_project_id ON voice_libraries(project_id)"))
conn.execute(sa.text("CREATE INDEX IF NOT EXISTS ix_voice_libraries_status ON voice_libraries(status)"))
def downgrade() -> None:
conn = op.get_bind()
# Drop new tables
conn.execute(sa.text("DROP TABLE IF EXISTS voice_libraries"))
conn.execute(sa.text("DROP TABLE IF EXISTS title_libraries"))
# Remove metadata columns
conn.execute(sa.text("ALTER TABLE generation_tasks DROP COLUMN IF EXISTS metadata"))
conn.execute(sa.text("ALTER TABLE edit_templates DROP COLUMN IF EXISTS metadata"))
conn.execute(sa.text("ALTER TABLE assets DROP COLUMN IF EXISTS metadata"))
conn.execute(sa.text("ALTER TABLE asset_libraries DROP COLUMN IF EXISTS metadata"))
conn.execute(sa.text("ALTER TABLE projects DROP COLUMN IF EXISTS metadata"))