4

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

使用 id 作为整数和主键添加表后，id 列将是一个自动增量列。如何查询升级脚本中的数据，以便确定我得到正确的 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')})
        )
4

1 回答 1

5

首先要拥有自动增量列，您需要修改表模式定义以为主键列传递序列：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 ...)您可以在相同的上下文中运行选择查询并检索感兴趣记录的 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 子句中指定适当的过滤器，以确保您以独特的方式识别了您最近更改的记录。

这是一些类似功能的示例，但它具有硬编码的选择查询

于 2013-03-21T18:37:06.117 回答