55 lines
1.3 KiB
Python
55 lines
1.3 KiB
Python
"""添加 task_id 字段到 Task 表
|
||
|
||
Revision ID: 001
|
||
Revises:
|
||
Create Date: 2023-07-01
|
||
|
||
"""
|
||
from alembic import op
|
||
import sqlalchemy as sa
|
||
from sqlalchemy import text
|
||
import uuid
|
||
from sqlalchemy.engine import reflection
|
||
import sys
|
||
import os
|
||
from pathlib import Path
|
||
|
||
# 添加 migrations 目录到 Python 路径
|
||
sys.path.insert(0, str(Path(__file__).parent.parent))
|
||
|
||
# 导入迁移辅助模块
|
||
from migration_helpers import (
|
||
safe_add_column,
|
||
safe_create_index,
|
||
safe_alter_column,
|
||
generate_uuid_for_null_values,
|
||
safe_drop_index,
|
||
safe_drop_column
|
||
)
|
||
|
||
# revision identifiers, used by Alembic.
|
||
revision = '001'
|
||
down_revision = None
|
||
branch_labels = None
|
||
depends_on = None
|
||
|
||
|
||
def upgrade():
|
||
# 添加 task_id 字段到 Task 表
|
||
if safe_add_column('tasks', 'task_id', sa.String(36), nullable=True, comment='任务UUID，用于外部引用'):
|
||
# 为已有记录生成 UUID
|
||
generate_uuid_for_null_values('tasks', 'task_id')
|
||
|
||
# 添加唯一索引
|
||
safe_create_index('ix_tasks_task_id', 'tasks', ['task_id'], unique=True)
|
||
|
||
# 添加非空约束
|
||
safe_alter_column('tasks', 'task_id', nullable=False, existing_type=sa.String(36))
|
||
|
||
|
||
def downgrade():
|
||
# 删除索引
|
||
safe_drop_index('ix_tasks_task_id', 'tasks')
|
||
|
||
# 删除列
|
||
safe_drop_column('tasks', 'task_id') |