python - 在 Python 中使用 psycopg2 获取插入行到 Redshift 表中的自动 ID

Question

我正在使用 psycopg2 库从 Python 2.7 将记录插入到 Amazon Redshift 表中，我想恢复自动生成插入行的主 ID。

我已经尝试了我可以在这里或使用谷歌搜索在其他网站上找到的常用方法，例如:

conn=psycopg2.connect(conn_str)
conn.autocommit = True

sql = "INSERT INTO schema.table (col1, col2) VALUES (%s, %s) RETURNING id;"

cur = conn.cursor()
cur.execute(sql,(val1,val2))
id = cur.fetchone()[0]

我在 cur.execute 行收到一个错误:

ProgrammingError: syntax error at or near "RETURNING"

有人知道如何解决这个问题或完成同样的事情吗？

I have to use psycopg2 in my code

Answer 1

目前无法使用 Redshift，因为它不支持通过 RETURNING 语法返回最后插入的 ID。您可能需要做的是在事务中使用 SELECT MAX(id) FROM schema.table;，这可能不是您想听到的，但似乎是您可以使用Redshift 的当前状态。