时间戳

首页 > 解决方案 > SQLAlchemy插入返回具有多个值的主键

python - SQLAlchemy插入返回具有多个值的主键

问题描述

我正在尝试使用带有字典列表作为值的 SQLAlchemy 的插入和执行方法，并取回插入的 ID。数据库是 Postgres，应该支持 RETURNING。

代码看起来或多或少是这样的：

table = MyThing.__table__
insert_stmt = table.insert().values({"name": bindparam("name_val")}).returning(table.c.id)
values = [{"name_val": "a"}, {"name_val": "b"}]
result = session.execute(insert_stmt, values).fetchall()

（为了可读性和一些混淆而进行了修剪）

当值列表大于一时（如上例所示），我得到：

sqlalchemy.exc.ProgrammingError: (psycopg2.ProgrammingError) 没有要获取的结果

但是当列表只有一个值时，新的 id 会正确返回。

我知道我可能在使用返回的方式上错过了一些东西，但我不知道是什么。

标签： pythonpostgresqlsqlalchemybulkinsert

解决方案

相同的绑定参数在您传递给的第二个参数的值列表中重复session.execute。

values插入多个值时应该是一个列表，并且默认情况下 sqlalchemy 绑定参数，除非literal_binds选项设置为，True以便在不显式绑定参数的情况下安全地做

values = [{"name_val": "a"}, {"name_val": "b"}]
insert_stmt = table.insert().values(values).returning(table.c.id)
result = session.execute(insert_stmt).fetchall()

假设您想保留现有的显式绑定， values为插入语句传递的必须是关键字参数。您现有的查询将更新如下：

bind_params = [{"name_val": "a"}, {"name_val": "b"}]
insert_stmt = table.insert().values(name_val=bindparam("name_val")).returning(table.c.id)
result = session.execute(insert_stmt, bind_params).fetchall()

推荐阅读