时间戳

首页 > 解决方案 > Psycopg2 用一个查询插入多行

python-3.x - Psycopg2 用一个查询插入多行

问题描述

我在 postgresql 中有带有字段id（唯一）和val. 我想执行这样的事情：

INSERT INTO my_table (id, val) 
VALUES (%(id)s, %(val)s)
ON CONFLICT(id) DO UPDATE
SET val = val + %(val)s

要插入的数据就像[{"id": 123, "val": 5}, {"id": 456, "val": 8}, ...]. 有没有办法通过一个查询来更新所有这些？

cursor.executemany不会，这与在一个循环中使用所有这些字典进行查询是一样的。如果没有 ON CONFLICT DO UPDATE 我可以做类似的事情"insert into mytable (id, val) values " + ', '.join(['(%s, %s)'] * len(data))并将数据转换到列表 [id1, val1, id2, val2, ...]。但我不知道如何组合多个值来插入和更新语句。

标签： python-3.xpsycopg2

解决方案

I have the same problem. After a while searching, I found two post:

  1. Posgresql - upsert: https://dba.stackexchange.com/questions/167591/postgresql-psycopg2-upsert-syntax-to-update-columns
  2. Psycopg2 - insert many rows: psycopg2: insert multiple rows with one query

=> So the answer for you is:

    # my table: cas(address, id, description) - address is primary key
data = [('0x18f9f00a432F50c6E2429d31776724d3cB873BEF', '1000', 'mot ngan'),
        ('0x06471C53CE649Eb4dA88b792D500544A7E5C9635', '2000', 'hai ngan')]
args = [cur.mogrify('(%s, %s, %s)', x).decode('utf-8')
        for x in data]
args_str = ', '.join(args)
cur.execute('''INSERT INTO cas (address, id, description) VALUES'''
            + args_str +
            '''ON CONFLICT (address) DO UPDATE SET     
        (address, id, description) = (EXCLUDED.address, EXCLUDED.id, EXCLUDED.description)''')

推荐阅读