Python/postgres/psycopg2 : getting ID of row just inserted

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 注入(inject):

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