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

105

Я использую 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)

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

Благодарность!

AP257
источник

Ответы:

214
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

ThiefMaster
источник
13
Чтобы уточнить, idin RETURNING idдолжно быть имя поля поля серийного / первичного ключа.
joshden 06
11
курсор fetchone дает мне "нет результатов для выборки".
Леонид
@ Леонид, ты понял это?
Alison S
6
@AlisonS @Leonid У меня была такая же ошибка, но добавление RETURNING idв конец INSERTзапроса исправило ее для меня.
Banjer
Возможно, это небольшое примечание, но важный момент, о котором стоит упомянуть для всех: убедитесь, что вы используете только курсор .execute (), а не курсор .mogrify () перед командой execute () , в противном случае (как в моем случае) cursor.fetchone () не даст результатов! Используя только курсор .execute () без «ничего» перед этой командой, вы получите идентификатор.
TheHeroOfTime,
14

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

cursor.execute ('ВСТАВИТЬ В ........')
cursor.execute ('ВЫБРАТЬ ЛАСТВАЛЬ ()')
lastid = cursor.fetchone () ['lastval']

На всякий случай кому было полезно!

Джейми Браун
источник
4
Просто помните - выполнение этого в двух подобных инструкциях сопряжено с (очень небольшим) риском состояния гонки, если что-то вставляет строку в базу данных непосредственно после вас, но до того, как ваша команда lastval () вернет текущее значение последовательности.
Дэйв Томас