Question

我正在使用alembic来管理我的数据库结构。

使用id作为Integer和主键添加表后，id列将是一个autoincrement-column。如何查询升级脚本中的数据，以便我确定我获得了正确的ID（在这种特定情况下我知道它是1）？

我知道如何

#creating the table
op.create_table(
    'srv_feed_return_type',
    sa.Column('id', sa.Integer, primary_key=True),
    sa.Column('name', sa.String(50), nullable=False),
    sa.Column('created', sa.DateTime, server_default=func.now(), nullable=False),
    sa.Column('created_by', sa.String(50), nullable=False),
    sa.Column('last_updated', sa.DateTime, nullable=False),
    sa.Column('last_updated_by', sa.String(50), nullable=False)
)

#table for operations
srv_feed_return_type = table('srv_feed_return_type',
                             column('name'),
                             column('created'),
                             column('created_by'),
                             column('last_updated'),
                             column('last_updated_by'))

#bulk insert
op.bulk_insert(srv_feed_return_type,
               [
                   {'name': 'dataset',
                    'created': datetime.now(), 'created_by': 'Asken',
                    'last_updated': datetime.now(), 'last_updated_by': 'Asken'}
               ])

我知道我可以更新，但如何使用类似下面的内容进行选择？

op.execute(
    srv_feed_return_type.update().\
        where(srv_feed_return_type.c.name==op.inline_literal('dataset')).\
        values({'name':op.inline_literal('somethingelse')})
        )

Answer 1

首先要有自动增量列，您需要修改表模式定义，以便为主键列传递Sequence：sa.Column('id', sa.Integer, Sequence('srv_feed_r_t_seq'),primary_key=True),

#creating the table
op.create_table(
    'srv_feed_return_type',
    sa.Column('id', sa.Integer, Sequence('srv_feed_r_t_seq'),primary_key=True),
    sa.Column('name', sa.String(50), nullable=False),
    sa.Column('created', sa.DateTime, server_default=func.now(), nullable=False),
    sa.Column('created_by', sa.String(50), nullable=False),
    sa.Column('last_updated', sa.DateTime, nullable=False),
    sa.Column('last_updated_by', sa.String(50), nullable=False)
)

现在关于如何获得PK id：

op.execute和op.bulk_insert不会向您返回任何结果。但是您可以获得用于这些操作的相同连接。

执行bulk_insert或execute(table.update ...)后，您可以在同一个上下文中运行select查询并检索感兴趣的记录的PK ID：

connection = op.get_bind()
r = connection.execute(srv_feed_return_type.select().where(...))
for row in r:
    pk_id = r['id']
    """or something more sophisticated"""

您需要在where子句中指定正确的过滤器，以确保您识别出最近以独特方式更改的记录。

以下是some example of similar functionality，但它有硬编码的选择查询

alembic并获取最后插入的值

1 个答案: