Как мне получить «id» после INSERT в базу данных MySQL с Python?

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

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

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

Моя таблица имеет 2 столбца:

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

Как я могу получить «идентификатор», после того, как я только что вставил это?

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

Также 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.

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

id = cursor.lastrowid

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

SELECT @@IDENTITY AS 'Identity';

или

SELECT last_insert_id();

Это может быть просто требованием PyMySql в Python, но я обнаружил, что мне нужно назвать точную таблицу, для которой я хотел получить идентификатор:

В:

cnx = pymysql.connect(host='host',
                            database='db',
                            user='user',
                            password='pass')
cursor = cnx.cursor()
update_batch = """insert into batch set type = "%s" , records = %i, started = NOW(); """
second_query = (update_batch % ( "Batch 1", 22  ))
cursor.execute(second_query)
cnx.commit()
batch_id = cursor.execute('select last_insert_id() from batch')
cursor.close()

batch_id

Out: 5
... или любое другое правильное значение Batch_ID