迭代返回execute_values

时间:2017-05-19 08:54:06

标签: psycopg2

http://initd.org/psycopg/docs/extras.html

psycopg2.extras.execute_values有一个参数page_size。 我正在使用INSERT INTO... ON CONFLICT...进行RETURNING ID

问题是,cursor.fetchall()只返回最后一个"页面",即100个ID(默认为page_size)。

如果不修改page_size参数,是否可以迭代结果,以更新总行数?

1 个答案:

答案 0 :(得分:0)

我遇到了同样的问题。我通过批量调用execute_values()解决此问题。我将设置my_page_size = 1000,然后遍历我的值,填充argslist直到有my_page_size项。然后,我将调用execute_values(cur,sql,argslist,page_size = my_page_size)。然后遍历cur以获得这些ID。