如何在使用Python在SQLite中插入行之後檢索插入的id？

Question

如何在使用Python在SQLite中插入行之後檢索插入的id？我的表是這樣的：

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

我插入新行帶有示例數據username="test"和password="test"。如何以交易安全的方式檢索生成的ID？這是一個網站解決方案，兩個人可能同時插入數據。我知道我可以得到最後一個讀取行，但我不認爲這是交易安全。有人可以給我一些建議嗎？

Answer 1

你可以使用cursor.lastrowid（請參閱「可選DB 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 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