如何在使用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