Python / postgres / psycopg2: получение идентификатора только что вставленной строки

Я использую Python и psycopg2 для интерфейса с postgres.

когда я вставляю строку...

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

... как мне получить идентификатор строки, которую я только что вставил? Попытка:

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)

может кто-нибудь посоветуете?

спасибо!

3 68
python postgresql psycopg2

3 ответа:

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

143

я оказался здесь, потому что у меня была аналогичная проблема, но мы используем Postgres-XC, который еще не поддерживает предложение RETURNING ID. В этом случае вы можете использовать:

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

на всякий случай это было полезно для всех!

9

рассмотрим предложение возврата http://www.postgresql.org/docs/8.3/static/sql-insert.html

2