Comment obtenir l '«id» après INSERT dans la base de données MySQL avec Python?

J'exécute une instruction INSERT INTO

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

et je veux obtenir la clé primaire.

Mon tableau comporte 2 colonnes:

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

Comment puis-je obtenir l '"identifiant" après avoir inséré ceci?

Utilisez cursor.lastrowidpour obtenir l'ID de la dernière ligne insérée sur l'objet curseur, ou connection.insert_id()pour obtenir l'ID de la dernière insertion sur cette connexion.

Aussi, cursor.lastrowid(une extension dbapi / PEP249 prise en charge par 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.lastrowidest un peu moins cher connection.insert_id()et beaucoup moins cher qu'un autre aller-retour vers MySQL.

La spécification Python DBAPI définit également l'attribut 'lastrowid' pour l'objet curseur, donc ...

id = cursor.lastrowid

... devrait fonctionner aussi, et c'est évidemment basé sur la connexion.

SELECT @@IDENTITY AS 'Identity';

ou

SELECT last_insert_id();

Cela pourrait être juste une exigence de PyMySql en Python, mais j'ai trouvé que je devais nommer la table exacte pour laquelle je voulais l'ID:

Dans:

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 quelle que soit la valeur Batch_ID correcte