如何通过Alembic迁移为MySQL现有表添加非主键自增列

嘿，我来帮你搞定这个Alembic迁移的问题！你已经有了正确的MySQL原生SQL语句，不过直接用Alembic的op.add_column可能会踩坑——因为SQLAlchemy的抽象层默认把autoincrement=True和主键绑定，而且没法直接指定FIRST（把列放在表的第一个位置）。下面给你两种靠谱的解决方案：

方案一：直接执行原生SQL（最简便） #

既然你已经验证过原生SQL是有效的，那直接在Alembic迁移脚本里用op.execute执行这条语句就完事了，完全绕过SQLAlchemy的限制，适配MySQL的特有语法：

from alembic import op

def upgrade():
    # 直接执行你的原生ALTER TABLE语句
    op.execute("ALTER TABLE `mytable` ADD `id` INT UNIQUE NOT NULL AUTO_INCREMENT FIRST")

def downgrade():
    # 回滚操作就是删除这个列
    op.drop_column('mytable', 'id')

这个方案的好处是简单直接，完全符合你的需求，不需要额外的分步操作，适合已经确认原生SQL没问题的场景。

方案二：用SQLAlchemy API分步实现（更贴近ORM风格） #

如果你想尽量用Alembic/SQLAlchemy的API来做，可以分三步操作，避开MySQL的限制：

from alembic import op
import sqlalchemy as sa

def upgrade():
    # 1. 先添加一个允许为空的临时列，暂不设置自增和非空约束
    op.add_column('mytable', sa.Column('id', sa.INTEGER(), nullable=True))
    
    # 2. 为现有数据填充递增的ID值（用MySQL变量实现自增赋值）
    op.execute("SET @row_number = 0;")
    op.execute("UPDATE `mytable` SET `id` = (@row_number := @row_number + 1);")
    
    # 3. 修改列属性：设置非空、唯一约束、自增，同时把列移到表的第一个位置
    op.execute("ALTER TABLE `mytable` MODIFY COLUMN `id` INT UNIQUE NOT NULL AUTO_INCREMENT FIRST")

def downgrade():
    op.drop_column('mytable', 'id')

注意事项： #

• 不管用哪种方案，迁移前一定要备份数据，避免操作失误导致数据丢失
• 如果你的表数据量很大，ALTER TABLE操作会锁表，尽量在业务低峰期执行
• MySQL规定一个表只能有一个AUTO_INCREMENT列，所以要确保你的表之前没有其他自增列

内容的提问来源于stack exchange，提问作者adrpino