sqlalchemy - 批量插入 sqlalchemy 核心与冲突更新

Question

我想将多个项目插入到一个表中，并在发生冲突时更新该表。这是我想出的以下内容

from sqlalchemy.dialects.postgresql import insert
meta = MetaData()
jobs_table = Table('jobs', meta, autoload=True, autoload_with=engine)
stmt = insert(jobs_table).values(jobs)
stmt.on_conflict_do_update(
    index_elements=['j_id'],
    set_= dict(active=True)
)
result = engine.execute(stmt)
return result.is_insert

j_id 是一个唯一字段，如果该行已存在，我将尝试对其进行更新。如果该行已存在，我会收到以下错误。

(psycopg2.IntegrityError) duplicate key value violates unique constraint "j_id"
DETAIL:  Key (j_id)=(0ea445da-bd1d-5571-9906-0694fa85728a) already exists.

我在这里遗漏了什么吗？

Answer 1

stmt.on_conflict_do_update 返回一个新语句。如果您更改为以下内容，它应该可以工作:

from sqlalchemy.dialects.postgresql import insert
meta = MetaData()
jobs_table = Table('jobs', meta, autoload=True, autoload_with=engine)
stmt = insert(jobs_table).values(jobs)
stmt = stmt.on_conflict_do_update(index_elements=['j_id'],
                                  set_= dict(active=True))
result = engine.execute(stmt)
return result.is_insert

您可以print(stmt) 语句来查看生成的 SQL 查询。这对于查看您要执行的语句是否具有预期的表达式很有用。将 echo=True 添加到 create_engine 也有助于检测问题!