Как я могу получить "идентификатор" после INSERT в базу данных MySQL с Python?

Question

Я выполняю инструкцию INSERT INTO

cursor.execute("INSERT INTO mytable(height) VALUES(%s)",(height))

и я хочу получить первичный ключ.

В моей таблице 2 столбца:

id      primary, auto increment
height  this is the other column.

Как я могу получить "id" после того, как я только что вставил это?

Answer 1

Используйте cursor.lastrowid для получения последнего идентификатора строки, вставленного в объект курсора, или connection.insert_id() для получения идентификатора из последней вставки этого соединения.

Answer 2

Также cursor.lastrowid (расширение dbapi / PEP249, поддерживаемое MySQLdb):

>>> import MySQLdb
>>> connection = MySQLdb.connect(user='root')
>>> cursor = connection.cursor()
>>> cursor.execute('INSERT INTO sometable VALUES (...)')
1L
>>> connection.insert_id()
3L
>>> cursor.lastrowid
3L
>>> cursor.execute('SELECT last_insert_id()')
1L
>>> cursor.fetchone()
(3L,)
>>> cursor.execute('select @@identity')
1L
>>> cursor.fetchone()
(3L,)

cursor.lastrowid несколько дешевле, чем connection.insert_id(), и намного дешевле, чем очередное путешествие в MySQL.

Answer 3

Спецификация Python DBAPI также определяет атрибут 'lastrowid' для объекта курсора, поэтому ...

id = cursor.lastrowid

... тоже должно работать, и это, очевидно, для каждого соединения.

Answer 4

SELECT @@IDENTITY AS 'Identity';

или

SELECT last_insert_id();