Як отримати "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.

Як отримати "id", після того як я тільки що вставив це?

Використовуйте 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

Вихід: 5
... або будь-яке правильне значення Batch_ID насправді