python - sqlalalchemy postgresql 插入语句忽略重复项

Question

我正在尝试使用 SQLAlchemy 将数据批量插入到 postgres，但无法找到如何通过唯一约束忽略重复值

我的代码

from sqlalchemy.dialects.postgresql import insert

stmt = insert(User, inline=True)

values = [{"id": 1, "name": "David"}, {"id": 2, "name": "Cris"}]

session.execute(stmt, values)
session.commit()

出现这样的错误

sqlalchemy.exc.IntegrityError: (psycopg2.IntegrityError) duplicate key value violates unique constraint "uq_users_name"

Answer 1

默认情况下，sqlalchemy 不支持ignore_duplicates 选项，因此我们需要通过简单地在末尾添加 ON CONFLICT DO NOTHING 字符串来向 sqlalchemy 查询编译器添加自定义方法查询

from sqlalchemy.dialects.postgresql import insert
from sqlalchemy.ext.compiler import compiles

@compiles(Insert, 'postgresql')
def ignore_duplicates(insert, compiler, **kw):
    s = compiler.visit_insert(insert, **kw)
    ignore = insert.kwargs.get('postgresql_ignore_duplicates', False)
    return s if not ignore else s + ' ON CONFLICT DO NOTHING'

Insert.argument_for('postgresql', 'ignore_duplicates', None)


stmt = insert(User, postgresql_ignore_duplicates=True, inline=True)

values = [{"id": 1, "name": "David"}, {"id": 2, "name": "Cris"}]

session.execute(stmt, values)
session.commit()