我正在尝试使用元组元组一次更新多行。
我想出了如何从this post构造sql语句,但在psycopg2
中实现它已经证明更具挑战性。
这就是我所拥有的:
c = db.cursor()
new_values = (("Richard",29),("Ronald",30))
sql = """UPDATE my_table AS t
SET name = e.name
FROM (VALUES %s) AS e(name, id)
WHERE e.id = t.id;"""
c.execute(sql, (new_values,))
结果是错误:ProgrammingError: table "e" has 1 columns available but 2 columns specified
这是因为FROM
子句被解释为:
FROM (VALUES (("Richard",29),("Ronald",30)))
而不是:
FROM (VALUES ("Richard",29),("Ronald",30))
我可以通过执行以下操作解决此问题但似乎不安全:
import re
c = db.cursor()
sql = """UPDATE my_table AS t
SET name = e.name
FROM (VALUES %s) AS e(name, id)
WHERE e.id = t.id;"""
sql = c.mogrify(sql, (new_values,))
# Replace the first occurance of '((' with '('.
sql = sql.replace('((', '(',1)
# Replace the last occurance of '))' with ')'.
sql = re.sub(r'(.*)\)\)', r'\1)', sql)
sql = c.execute(sql)
有更好的方法吗?
答案 0 :(得分:10)
This post指出了我正确的方向。 extras.execute_values
的{{3}}也包含使用UPDATE
子句的一个很好的示例。
c = db.cursor()
update_query = """UPDATE my_table AS t
SET name = e.name
FROM (VALUES %s) AS e(name, id)
WHERE e.id = t.id;"""
psycopg2.extras.execute_values (
c, update_query, new_values, template=None, page_size=100
)