如何在Python中使用SQLite插入行后获取插入的ID？

Question

如何在使用Python插入行后从SQLite中获取插入的ID？我有一个这样的表：

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

我插入了一行新的数据，比如 username="test" 和 password="test"。我该如何以安全的方式获取生成的ID呢？这是为了一个网站的解决方案，可能会有两个人同时插入数据。我知道可以获取最后一行的ID，但我觉得那样不够安全。有人能给我一些建议吗？

Answer 1

所有的感谢都要给评论区的@Martijn Pieters：

你可以使用这个函数last_insert_rowid()：

last_insert_rowid()这个函数会返回最后一次插入的行的ROWID，这个行是通过调用这个函数的数据库连接插入的。last_insert_rowid()这个SQL函数其实是对sqlite3_last_insert_rowid()这个C/C++接口函数的封装。

SQLite 3.35的RETURNING子句：

CREATE TABLE users (
  id INTEGER PRIMARY KEY,
  first_name TEXT,
  last_name TEXT
);

INSERT INTO users (first_name, last_name)
VALUES ('Jane', 'Doe')
RETURNING id;

在INSERT、UPDATE和DELETE语句中返回插入行的请求列。Python的用法：

cursor.execute('INSERT INTO users (first_name, last_name) VALUES (?,?)'
               ' RETURNING id',
               ('Jane', 'Doe'))
row = cursor.fetchone()
(inserted_id, ) = row if row else None

Answer 2

你可以使用 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 会返回那个 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