Como obtenho o "id" após INSERT no banco de dados MySQL com Python?

Eu executo uma instrução INSERT INTO

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

e quero obter a chave primária.

Minha tabela possui 2 colunas:

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

Como obtenho o "id" depois de inseri-lo?

Use cursor.lastrowidpara obter o último ID da linha inserido no objeto cursor, ou connection.insert_id()para obter o ID da última inserção nessa conexão.

Além disso, cursor.lastrowid(uma extensão dbapi / PEP249 suportada pelo 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é um pouco mais barato que connection.insert_id()e muito mais barato que outra viagem de ida e volta ao MySQL.

A especificação DBAPI do Python também define o atributo 'lastrowid' para o objeto cursor, então ...

id = cursor.lastrowid

... deve funcionar também, e é obviamente baseado em cada conexão.

SELECT @@IDENTITY AS 'Identity';

ou

SELECT last_insert_id();

Isso pode ser apenas um requisito do PyMySql no Python, mas descobri que precisava nomear a tabela exata para a qual queria o ID:

No:

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
... ou qualquer que seja o valor correto de Batch_ID