Jak uzyskać „id” po INSERT w bazie danych MySQL za pomocą Pythona?

Wykonuję instrukcję INSERT INTO

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

i chcę zdobyć klucz podstawowy.

Moja tabela ma 2 kolumny:

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

Jak uzyskać „identyfikator” po tym, jak właśnie go wstawiłem?

Użyj, cursor.lastrowidaby wstawić identyfikator ostatniego wiersza do obiektu kursora lub connection.insert_id()aby uzyskać identyfikator z ostatniej wstawki w tym połączeniu.

Ponadto cursor.lastrowid(rozszerzenie dbapi / PEP249 obsługiwane przez 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.lastrowidjest nieco tańszy connection.insert_id()i znacznie tańszy niż kolejna podróż w obie strony do MySQL.

Specyfikacja DBAPI w języku Python definiuje również atrybut „lastrowid” dla obiektu kursora, więc ...

id = cursor.lastrowid

... powinien również działać, i oczywiście jest oparty na połączeniu.

SELECT @@IDENTITY AS 'Identity';

lub

SELECT last_insert_id();

Może to być tylko wymóg PyMySql w Pythonie, ale okazało się, że muszę podać dokładną tabelę, dla której chciałbym mieć identyfikator:

W:

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
... lub cokolwiek to jest poprawna wartość Batch_ID