Python/postgres/psycopg2：获取刚刚插入的行的ID

Question

我正在使用 Python 和 psycopg2 来连接 postgres。

当我插入一行时...

sql_string = "INSERT INTO hundred (name,name_slug,status) VALUES ("
sql_string += hundred_name + ", '" + hundred_slug + "', " + status + ");"
cursor.execute(sql_string)

...如何获取刚刚插入的行的 ID？尝试：

hundred = cursor.fetchall() 

返回错误，而使用RETURNING id：

sql_string = "INSERT INTO domes_hundred (name,name_slug,status) VALUES ("
sql_string += hundred_name + ", '" + hundred_slug + "', " + status + ") RETURNING id;"
hundred = cursor.execute(sql_string)

仅返回None。

更新： currval 也是如此（即使直接在 postgres 中使用此命令也可以）：

sql_string = "SELECT currval(pg_get_serial_sequence('hundred', 'id'));"
hundred_id = cursor.execute(sql_string)

任何人都可以建议吗？

谢谢！

Answer 1

cursor.execute("INSERT INTO .... RETURNING id")
id_of_new_row = cursor.fetchone()[0]

并且请不要手动构建包含值的 SQL 字符串。您可以（并且应该！）单独传递值，从而无需转义并且无法进行 SQL 注入：

sql_string = "INSERT INTO domes_hundred (name,name_slug,status) VALUES (%s,%s,%s) RETURNING id;"
cursor.execute(sql_string, (hundred_name, hundred_slug, status))
hundred = cursor.fetchone()[0]

请参阅 psycopg 文档以获取更多详细信息：http://initd.org/psycopg/docs/usage.html#passing-parameters-to-sql-queries

Answer 2

我最终来到这里是因为我遇到了类似的问题，但我们使用的是 Postgres-XC，它尚不支持 RETURNING ID 子句。在这种情况下，您可以使用：

cursor.execute('INSERT INTO ........')
cursor.execute('SELECT LASTVAL()')
lastid = cursor.fetchone()['lastval']

以防万一它对任何人都有用！

Answer 3

考虑 RETURNING 子句 http://www.postgresql.org/docs/8.3 /static/sql-insert.html

Answer 4

对我来说，盗贼大师的答案和杰米·布朗的答案都不起作用。对我有用的是两者的结合，我想在这里回答，以便它可以帮助其他人。

我需要做的是：

cursor.execute('SELECT LASTVAL()')
id_of_new_row = cursor.fetchone()[0]

语句 lastid = curve.fetchone()['lastval'] 对我不起作用，即使在 cursor.execute('SELECT LASTVAL()' 之后）。单独的语句 id_of_new_row = curve.fetchone()[0] 也不起作用。

也许我错过了一些东西。

Answer 5

ThiefMaster 的方法对我来说很有效，适用于 INSERT 和 UPDATE 命令。

如果在执行 INSERT/UPDATE 命令后在游标上调用 cursor.fetchone() 但缺少返回值 (RETURNING 子句）将引发异常：ProgrammingError('no results to fetch'))

insert_query = """
    INSERT INTO hundred (id, name, name_slug, status)
        VALUES (DEFAULT, %(name)s, %(name_slug)s, %(status)s)
        RETURNING id;
"""

insert_query_values = { 
    "name": "",
    "name_slug": "",
    "status": ""
}

connection = psycopg2.connect(host="", port="", dbname="", user="", password="")

try:
    with connection:
        with connection.cursor() as cursor:
            cursor.execute(insert_query, insert_query_values)
            num_of_rows_affected = cursor.rowcount
            new_row_id = cursor.fetchone()

except psycopg2.ProgrammingError as ex:
    print("...", ex)
    raise ex

finally:
    connection.commit()
    connection.close()