使用SQLAlchemy Postgres进行批量Upsert

Question

使用SQLAlchemy Postgres进行批量Upsert

12

我正在按照SQLAlchemy文档这里的指引撰写一个批量更新和插入(Postgres中称为upsert) 的语句。为了演示，我有一个简单的表MyTable:

class MyTable(base):
    __tablename__ = 'mytable'
    id = Column(types.Integer, primary_key=True)
    test_value = Column(types.Text)

创建一个通用的插入语句很简单：

from sqlalchemy.dialects import postgresql

values = [{'id': 0, 'test_value': 'a'}, {'id': 1, 'test_value': 'b'}]
insert_stmt = postgresql.insert(MyTable.__table__).values(values)

我遇到的问题是在尝试添加upsert的"on conflict"部分时。

update_stmt = insert_stmt.on_conflict_do_update(
    index_elements=[MyTable.id],
    set_=dict(data=values)
)

尝试执行此语句会导致ProgrammingError：

from sqlalchemy import create_engine
engine = create_engine('postgres://localhost/db_name')

engine.execute(update_stmt)

>>> ProgrammingError: (psycopg2.ProgrammingError) can't adapt type 'dict'

我认为我的误解在于使用on_conflict_do_update方法构建语句。有没有人知道如何构建这个语句？我已经查看了StackOverflow上的其他问题（例如这里），但似乎找不到解决以上错误的方法。

- user144153

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Paul Becotte · Accepted Answer

update_stmt = insert_stmt.on_conflict_do_update(
    index_elements=[MyTable.id],
    set_=dict(data=values)
)

index_elements应该是一个字符串列表或列对象的列表。因此可以是[MyTable.id]或['id']（这是正确的）。

set_应该是一个以列名为键，有效的SQL更新对象为值的字典。您可以使用excluded属性引用插入块中的值。所以在这里得到你所希望的结果的话，你需要 set_ = {'test_value': insert_stmt.excluded.test_value} （你的错误在于在例子中data=不是一个特殊参数...它是他们的示例表格上的列名）。

因此，整个内容如下：

update_stmt = insert_stmt.on_conflict_do_update(
    index_elements=[MyTable.id],
    set_={'test_value': insert_stmt.excluded.test_value}
)

当然，在实际的世界例子中，我通常想改变不止一列。在这种情况下，我会这样做...

update_columns = {col.name: col for col in insert_stmt.excluded if col.name not in ('id', 'datetime_created')}
update_statement = insert_stmt.on_conflict_do_update(index_elements=['id'], set_=update_columns)

（此示例将覆盖除id和datetime_created列以外的所有列）