¿Cómo obtengo el "id" después de INSERTAR en la base de datos MySQL con Python?

Ejecuto una instrucción INSERT INTO

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

y quiero obtener la clave primaria.

Mi mesa tiene 2 columnas:

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

¿Cómo obtengo el "id" después de insertar esto?

Use cursor.lastrowidpara obtener la última ID de fila insertada en el objeto del cursor, o connection.insert_id()para obtener la ID de la última inserción en esa conexión.

Además, cursor.lastrowid(una extensión dbapi / PEP249 compatible con 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.lastrowides algo más barato connection.insert_id()y mucho más barato que otro viaje de ida y vuelta a MySQL.

La especificación DBAPI de Python también define el atributo 'lastrowid' para el objeto cursor, así que ...

id = cursor.lastrowid

... también debería funcionar, y obviamente se basa en la conexión.

SELECT @@IDENTITY AS 'Identity';

o

SELECT last_insert_id();

Esto podría ser solo un requisito de PyMySql en Python, pero descubrí que tenía que nombrar la tabla exacta para la que deseaba la ID:

En:

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

Fuera: 5
... o cualquiera que sea el valor correcto de Batch_ID