PostgreSQL
 sql >> Base de Dados >  >> RDS >> PostgreSQL

Python/postgres/psycopg2:obtendo o ID da linha recém inserida

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

E por favor, não construa strings SQL contendo valores manualmente. Você pode (e deve!) passar valores separadamente, tornando desnecessário o escape e a injeção de SQL impossível:
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]

Consulte os documentos do psycopg para obter mais detalhes:http://initd.org/psycopg/docs/usage.html#passing-parameters-to-sql-queries