python-3.x - 是否有一种 python-alembic 方法可以在删除和添加列之间转换数据？

Question

我有一个 sqlite3 数据库，使用 python3 中的 SQLAlchemy 访问它。我想使用数据库迁移工具 alembic 添加一个新列并删除一个旧列。简单示例:

class Model(_Base):
  __tablename__ = 'Model'
  _oid = Column('oid', sa.Integer, primary_key=True)
  _number_int = sa.Column('number_int', sa.Integer)

迁移后应该是这样的:

class Model(_Base):
  __tablename__ = 'Model'
  _oid = Column('oid', sa.Integer, primary_key=True)
  _number_str = sa.Column('number_str', sa.String(length=30))

这里的相关点是 _number_int 中的数据应该像这样转换成 _number_str:

number_conv = {1: 'one', 2: 'two', 3: 'three'}
_number_str = number_conv[_number_int]

是否有一种蒸馏器方法来解决这个问题？这意味着 alembic 本身是否会在其概念/设计中处理此类情况？我想知道我是否可以为此使用 alembic 工具，或者我是否必须为此编写自己的额外代码。

当然，原始数据转换起来稍微复杂一些。这只是这里的一个例子。

Answer 1

这是蒸馏器 operation reference .有一种方法叫做 bulk_insert()用于批量插入内容，但不用于迁移现有内容。似乎蒸馏器没有内置。但是您可以自己实现数据迁移。

文章 "Migrating content with alembic" 中描述了一种可能的方法.您需要在迁移文件中定义中间表，其中包含两列(number_int 和 number_str):

import sqlalchemy as sa

model_helper = sa.Table(
    'Model',
    sa.MetaData(),
    sa.Column('oid', sa.Integer, primary_key=True),
    sa.Column('number_int', sa.Integer),
    sa.Column('number_str', sa.String(length=30)),
)

并使用这个中间表将数据从旧列迁移到新列:

from alembic import op


def upgrade():
    # add the new column first
    op.add_column(
        'Model',
        sa.Column(
            'number_str',
            sa.String(length=30),
            nullable=True
        )
    )

    # build a quick link for the current connection of alembic
    connection = op.get_bind()

    # at this state right now, the old column is not deleted and the
    # new columns are present already. So now is the time to run the
    # content migration. We use the connection to grab all data from
    # the table, convert each number and update the row, which is 
    # identified by its id
    number_conv = {1: 'one', 2: 'two', 3: 'three'}
    for item in connection.execute(model_helper.select()):
        connection.execute(
            model_helper.update().where(
                model_helper.c.id == item.id
            ).values(
                number_str=number_conv[item.number_int]
            )
        )

    # now that all data is migrated we can just drop the old column
    # without having lost any data
    op.drop_column('Model', 'number_int')

这种方法有点嘈杂(您需要手动定义表)，但它确实有效。