如何使用Python在SQLite插入行后检索插入的ID？

如何使用Python在SQLite插入行后检索插入的ID？ 我有这样的桌子：

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

我用示例数据username="test"和password="test"插入新行。 如何以交易安全的方式检索生成的ID？ 这是一个网站解决scheme，两个人可能同时插入数据。 我知道我可以得到最后一个读取行，但我不认为这是交易安全。 有人可以给我一些build议吗？

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 s， cursor.lastrowid将返回cursor插入的最后一行的id ：

 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 

请注意，当您使用executemany一次插入多行时， lastrowid将返回None 。

 cursor.executemany('INSERT INTO foo (username,password) VALUES (?,?)', (('baz','bar'),('bing','bop'))) print(cursor.lastrowid) # None