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

Question

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

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

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

Answer 1

Вы можете использовать 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

Если два человека вставляют одновременно, если они используют разные cursor с, 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