Flask-Migrate/Alembic：如何为PostgreSQL现有表添加identity列？

问题背景 #

我尝试创建迁移，为现有表添加一个新的IDENTITY列，该列最终将作为表的新主键。

模型代码如下：

class Action(db.Model):
    id = db.Column(db.Integer(), db.Identity(), primary_key=True)  # 新主键
    uuid = db.Column(sqlalchemy_utils.UUIDType, index=True, nullable=False)  # 旧主键

使用flask db migrate生成迁移时，仅得到一个普通整数列的添加操作：

def upgrade():
    """Upgrade from previous version."""
    op.add_column("actions", sa.Column("id", sa.Integer(), nullable=False))

对于基于序列的ID列，我知道需要显式创建序列并设置server_default：

def upgrade():
    """Upgrade from previous version."""
    op.execute(sa.schema.CreateSequence(sa.Sequence("actions_id_seq")))
    op.add_column("actions", sa.Column("id", sa.Integer(), server_default=sa.text("nextval('actions_id_seq'::regclass)"), nullable=False))

当前使用SQLAlchemy 1.4.42和Alembic 1.8.1，找不到自动生成带GENERATED BY DEFAULT AS IDENTITY属性的IDENTITY列迁移的方法，求实现建议。

可行实现方案 #

方案一：执行原生DDL语句 #

直接通过原生SQL语句添加IDENTITY列，这种方式最直接，适配对应数据库的语法：

def upgrade():
    # 以PostgreSQL为例，其他数据库需调整语法
    op.execute("ALTER TABLE actions ADD COLUMN id INTEGER GENERATED BY DEFAULT AS IDENTITY PRIMARY KEY")

def downgrade():
    op.drop_column("actions", "id")

方案二：手动构造带Identity属性的列定义 #

利用SQLAlchemy的Identity对象手动定义列，让Alembic生成正确的IDENTITY约束：

from sqlalchemy import Identity

def upgrade():
    op.add_column(
        "actions",
        sa.Column(
            "id",
            sa.Integer(),
            Identity(always=False, start=1),  # always=False对应GENERATED BY DEFAULT
            nullable=False,
            primary_key=True
        )
    )

def downgrade():
    op.drop_column("actions", "id")

关键注意事项 #

• 不同数据库的IDENTITY语法有差异：MySQL用AUTO_INCREMENT，SQL Server用IDENTITY(1,1)，PostgreSQL用GENERATED ... AS IDENTITY，需根据实际使用的数据库调整代码。
• 若要将新列设为主键，需确保旧主键的约束可以安全移除，建议先完成数据迁移验证后再切换主键。
• 执行迁移前务必备份数据表，避免操作失误导致数据丢失。

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