使用Alembic执行更新查询时避免模型依赖，防止后续迁移冲突

问题根源 #

在迁移脚本中使用session.query(MyTable)时，SQLAlchemy会根据当前模型定义生成包含所有字段的SELECT语句。当未来向MyTable添加新字段（如foo）后，在未更新数据库的旧环境中执行该迁移时，数据库里还不存在foo字段，就会触发Unknown column 'my_table.foo'错误。

而若仅查询部分字段（如session.query(MyTable.last_name, MyTable.first_name)），返回的是命名元组/普通元组而非模型实例，因此无法直接设置属性，导致can't set attribute错误。

解决方案 #

方案1：使用load_only加载必要字段（推荐）

通过sqlalchemy.orm.load_only指定仅加载迁移所需的字段，既保留ORM实例的可修改性，又避免查询未创建的字段。

from sqlalchemy.orm import load_only

# 添加last_name字段（先设为nullable=True，方便填充数据）
op.add_column('my_table', sa.Column('last_name', sa.String(length=100), nullable=True)) 

# 仅加载主键index和需要处理的first_name
records = session.query(MyTable).options(load_only('index', 'first_name')).filter(MyTable.age == 0).all()

for record in records:
    record.last_name = do_some_processing(record.first_name)

session.commit()

# 数据填充完成后，将字段改为nullable=False（如果业务需要）
op.alter_column('my_table', 'last_name', nullable=False)

方案2：批量SQL更新（适合逻辑可通过SQL实现的场景）

若do_some_processing的逻辑可以用SQL函数实现，直接执行批量更新语句，无需查询实例，效率更高且完全脱离模型字段依赖。

from sqlalchemy import func

op.add_column('my_table', sa.Column('last_name', sa.String(length=100), nullable=True)) 

# 示例：用SQL反转first_name作为last_name（替换为你的实际逻辑）
op.execute(
    sa.update(MyTable)
    .where(MyTable.age == 0)
    .values(last_name=func.reverse(MyTable.first_name))
)

session.commit()
op.alter_column('my_table', 'last_name', nullable=False)

方案3：使用原始SQL操作（彻底脱离ORM依赖）

直接用原始SQL查询和更新，完全不依赖模型定义，适用于复杂的Python处理逻辑。

op.add_column('my_table', sa.Column('last_name', sa.String(length=100), nullable=True)) 

# 查询需要处理的数据
results = session.execute(
    sa.text("SELECT `index`, first_name FROM my_table WHERE age = :age"),
    {"age": 0}
).fetchall()

# 逐个处理并更新
for idx, first_name in results:
    processed_last_name = do_some_processing(first_name)
    session.execute(
        sa.text("UPDATE my_table SET last_name = :ln WHERE `index` = :idx"),
        {"ln": processed_last_name, "idx": idx}
    )

session.commit()
op.alter_column('my_table', 'last_name', nullable=False)

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