Files
xiaoxia-saas/alembic/versions/015_add_generation_task_extensions.py
T
Audit Bot 7783f36172
CI/CD Pipeline / Validate Code Quality And Tests (pull_request) Has been cancelled
CI/CD Pipeline / Frontend Lint (pull_request) Has been cancelled
fix: 从迁移 015 中移除残留的 edit_plan_id(代码审计 P1)
迁移 011 已删除 edit_plan_id 列,迁移 015 不应再 add_column。
同时从 downgrade() 中移除对应的 drop_column。
2026-06-29 20:23:23 +08:00

35 lines
1.2 KiB
Python

"""add generation task extensions
Revision ID: 015
Revises: 014
Create Date: 2026-06-29
"""
import sqlalchemy as sa
from sqlalchemy.dialects import mysql
from alembic import op
# revision identifiers
revision = "015"
down_revision = "014"
branch_labels = None
depends_on = None
def upgrade() -> None:
op.add_column("generation_tasks", sa.Column("template_id", sa.String(36), nullable=False, server_default=""))
op.add_column("generation_tasks", sa.Column("asset_ids", mysql.JSON(), nullable=False, server_default="[]"))
op.add_column("generation_tasks", sa.Column("title_ids", mysql.JSON(), nullable=False, server_default="[]"))
op.add_column("generation_tasks", sa.Column("voice_ids", mysql.JSON(), nullable=False, server_default="[]"))
op.create_index(op.f("ix_generation_tasks_template_id"), "generation_tasks", ["template_id"])
def downgrade() -> None:
op.drop_index(op.f("ix_generation_tasks_template_id"), table_name="generation_tasks")
op.drop_column("generation_tasks", "voice_ids")
op.drop_column("generation_tasks", "title_ids")
op.drop_column("generation_tasks", "asset_ids")
op.drop_column("generation_tasks", "template_id")