119 lines
3.2 KiB
Python
119 lines
3.2 KiB
Python
|
"""增加字段
|
|||
|
|
|
|||
|
|
Revision ID: 81b754397c49
|
|||
|
|
Revises: 001
|
|||
|
|
Create Date: 2025-03-14 18:01:04.421172
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
from alembic import op
|
|||
|
|
import sqlalchemy as sa
|
|||
|
|
from sqlalchemy import text
|
|||
|
|
from sqlalchemy.engine import reflection
|
|||
|
|
import logging
|
|||
|
|
from sqlalchemy.dialects import mysql
|
|||
|
|
|
|||
|
|
# 配置日志
|
|||
|
|
logging.basicConfig(level=logging.INFO)
|
|||
|
|
logger = logging.getLogger('alembic.migration')
|
|||
|
|
|
|||
|
|
# revision identifiers, used by Alembic.
|
|||
|
|
revision = '81b754397c49'
|
|||
|
|
down_revision = '001'
|
|||
|
|
branch_labels = None
|
|||
|
|
depends_on = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def safe_execute(func, *args, **kwargs):
|
|||
|
|
"""
|
|||
|
|
安全执行函数，捕获异常并记录日志
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
func: 要执行的函数
|
|||
|
|
*args: 位置参数
|
|||
|
|
**kwargs: 关键字参数
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
执行结果或 None（如果发生异常）
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
return func(*args, **kwargs)
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"执行 {func.__name__} 时发生错误: {e}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def column_exists(table_name, column_name):
|
|||
|
|
"""
|
|||
|
|
检查列是否存在
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
table_name: 表名
|
|||
|
|
column_name: 列名
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
bool: 列是否存在
|
|||
|
|
"""
|
|||
|
|
conn = op.get_bind()
|
|||
|
|
inspector = reflection.Inspector.from_engine(conn)
|
|||
|
|
columns = inspector.get_columns(table_name)
|
|||
|
|
return column_name in [column['name'] for column in columns]
|
|||
|
|
|
|||
|
|
|
|||
|
|
def index_exists(table_name, index_name):
|
|||
|
|
"""
|
|||
|
|
检查索引是否存在
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
table_name: 表名
|
|||
|
|
index_name: 索引名
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
bool: 索引是否存在
|
|||
|
|
"""
|
|||
|
|
conn = op.get_bind()
|
|||
|
|
inspector = reflection.Inspector.from_engine(conn)
|
|||
|
|
indexes = inspector.get_indexes(table_name)
|
|||
|
|
return index_name in [index['name'] for index in indexes]
|
|||
|
|
|
|||
|
|
|
|||
|
|
def table_exists(table_name):
|
|||
|
|
"""
|
|||
|
|
检查表是否存在
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
table_name: 表名
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
bool: 表是否存在
|
|||
|
|
"""
|
|||
|
|
conn = op.get_bind()
|
|||
|
|
inspector = reflection.Inspector.from_engine(conn)
|
|||
|
|
return table_name in inspector.get_table_names()
|
|||
|
|
|
|||
|
|
|
|||
|
|
def upgrade():
|
|||
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|||
|
|
with op.batch_alter_table('tasks', schema=None) as batch_op:
|
|||
|
|
batch_op.add_column(sa.Column('task_id', sa.String(length=36), nullable=False, comment='任务UUID，用于外部引用'))
|
|||
|
|
batch_op.alter_column('task_type',
|
|||
|
|
existing_type=mysql.ENUM('NORMAL', 'SCHEDULED', collation='utf8mb4_unicode_ci'),
|
|||
|
|
type_=sa.String(length=100),
|
|||
|
|
existing_comment='任务类型',
|
|||
|
|
existing_nullable=False)
|
|||
|
|
batch_op.create_index(batch_op.f('ix_tasks_task_id'), ['task_id'], unique=True)
|
|||
|
|
|
|||
|
|
# ### end Alembic commands ###
|
|||
|
|
|
|||
|
|
|
|||
|
|
def downgrade():
|
|||
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|||
|
|
with op.batch_alter_table('tasks', schema=None) as batch_op:
|
|||
|
|
batch_op.drop_index(batch_op.f('ix_tasks_task_id'))
|
|||
|
|
batch_op.alter_column('task_type',
|
|||
|
|
existing_type=sa.String(length=100),
|
|||
|
|
type_=mysql.ENUM('NORMAL', 'SCHEDULED', collation='utf8mb4_unicode_ci'),
|
|||
|
|
existing_comment='任务类型',
|
|||
|
|
existing_nullable=False)
|
|||
|
|
batch_op.drop_column('task_id')
|
|||
|
|
|
|||
|
|
# ### end Alembic commands ###
|