Как получить вставленный идентификатор после вставки строки в SQLite с помощью Python?

176

Как получить вставленный идентификатор после вставки строки в SQLite с помощью Python? У меня есть таблица, как это:

id INT AUTOINCREMENT PRIMARY KEY,
username VARCHAR(50),
password VARCHAR(50)

Я вставляю новую строку с данными примера username="test"иpassword="test" . Как получить сгенерированный идентификатор безопасным способом транзакции? Это для решения веб-сайта, где два человека могут вставлять данные одновременно. Я знаю, что могу получить последнюю прочитанную строку, но не думаю, что это безопасно для транзакций. Может кто-нибудь дать мне совет?

Джейн
источник

Ответы:

256

Вы можете использовать cursor.lastrowid (см. «Дополнительные расширения API БД»):

connection=sqlite3.connect(':memory:')
cursor=connection.cursor()
cursor.execute('''CREATE TABLE foo (id integer primary key autoincrement ,
                                    username varchar(50),
                                    password varchar(50))''')
cursor.execute('INSERT INTO foo (username,password) VALUES (?,?)',
               ('test','test'))
print(cursor.lastrowid)
# 1

Если два человека вставляют одновременно, если они используют разные cursors, cursor.lastrowidвернет idдля последней cursorвставленной строки :

cursor.execute('INSERT INTO foo (username,password) VALUES (?,?)',
               ('blah','blah'))

cursor2=connection.cursor()
cursor2.execute('INSERT INTO foo (username,password) VALUES (?,?)',
               ('blah','blah'))

print(cursor2.lastrowid)        
# 3
print(cursor.lastrowid)
# 2

cursor.execute('INSERT INTO foo (id,username,password) VALUES (?,?,?)',
               (100,'blah','blah'))
print(cursor.lastrowid)
# 100

Обратите внимание, что lastrowidвозвращается, Noneкогда вы вставляете более одной строки одновременно с executemany:

cursor.executemany('INSERT INTO foo (username,password) VALUES (?,?)',
               (('baz','bar'),('bing','bop')))
print(cursor.lastrowid)
# None
unutbu
источник
16
+1 за объяснение того, что он вернет самый последний идентификатор этого отдельного экземпляра курсора.
Quakkels
43
Существует также last_insert_rowid()функция SQL , позволяющая вставить идентификатор последней строки в качестве внешнего ключа в следующую инструкцию вставки, полностью в SQL.
Мартин Питерс
2
@MartijnPieters Вы можете опубликовать это как ответ, хотя вопрос довольно старый. Это все еще потенциально может помочь пользователям, читающим эту страницу.
Даниэль Вернер
3
Похоже, что Python sqlite3 lastrowid использует last_insert_rowidпод github.com/ghaering/pysqlite/blob/… (который не гарантированно безопасен, но кажется единственным вариантом FWIW). См. Также stackoverflow.com/q/2127138/32453
rogerdpack
2
Кстати, это работает, INSERTно не работает для UPDATE. Но при обновлении строки у вас, вероятно, уже есть соответствующий идентификатор строки (мне просто понадобилось время, чтобы понять это ...).
CGFoX
4

Все кредиты @Martijn Pieters в комментариях:

Вы можете использовать функцию last_insert_rowid():

last_insert_rowid()Функция возвращает ROWIDпоследнюю строку вставки из соединения с базой данных , которая вызывается функция. Функция last_insert_rowid()SQL является оболочкой для sqlite3_last_insert_rowid()интерфейсной функции C / C ++.

Стефан ван ден Аккер
источник