Как получить вставленный идентификатор после вставки строки в SQLite с помощью Python? У меня есть таблица вроде этого:
id INT AUTOINCREMENT PRIMARY KEY,
username VARCHAR(50),
password VARCHAR(50)
Я вставляю новую строку с примерами данных username="test"
и password="test"
. Как получить сгенерированный идентификатор безопасным способом? Это решение для веб-сайта, в котором два человека могут одновременно вставлять данные. Я знаю, что могу получить последнюю прочитанную строку, но я не думаю, что это безопасная транзакция. Может ли кто-нибудь дать мне совет?
Вы можете использовать cursor.lastrowid (см. "Дополнительные расширения API DB" ):
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
s, 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
last_insert_rowid()
SQLlast_insert_rowid()
, позволяющая вставлять идентификатор последней строки в качестве внешнего ключа в следующую инструкцию вставки, полностью в SQL.