如何使用Python在SQLite中插入行后检索插入的id?

i2byvkas  于 2022-11-15  发布在  SQLite
关注(0)|答案(2)|浏览(186)

如何使用Python在SQLite中插入行后检索插入的id?我的table是这样的:

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

我插入了一个包含示例数据username="test"password="test"的新行。如何以事务安全的方式检索生成的ID?这是一个网站解决方案,其中可能有两个人同时插入数据。我知道我可以获得最后读取的行,但我认为这不是事务安全。有人能给我一些建议吗?

kcrjzv8t

kcrjzv8t1#

您可以使用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将返回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
zvms9eto

zvms9eto2#

评论中的所有字幕都归功于@martjn Pieters:
您可以使用函数last_insert_rowid()
last_insert_rowid()函数从调用该函数的数据库连接返回最后插入的行的ROWIDlast_insert_rowid() SQL函数是sqlite3_last_insert_rowid() C/C++接口函数的 Package 器。

SQLite 3.35RETURNING子句:

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;

INSERTUPDATEDELETE语句中返回插入行的请求列。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

相关问题