วิธีการดึง id ที่ใส่เข้าไปหลังจากแทรกแถวใน SQLite โดยใช้ Python

วิธีการดึง id ที่ใส่เข้าไปหลังจากแทรกแถวใน SQLite โดยใช้ Python ฉันมีโต๊ะแบบนี้:

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

ผมแทรกแถวใหม่ที่มีข้อมูลตัวอย่างและusername="test" password="test"ฉันจะดึงข้อมูลรหัสที่สร้างขึ้นในวิธีการทำธุรกรรมได้อย่างปลอดภัย นี่เป็นวิธีการแก้ปัญหาเว็บไซต์โดยที่คนสองคนอาจแทรกข้อมูลในเวลาเดียวกัน ฉันรู้ว่าฉันสามารถอ่านแถวสุดท้ายได้ แต่ฉันไม่คิดว่าธุรกรรมนั้นปลอดภัย ใครสามารถให้คำแนะนำได้บ้าง

คุณสามารถใช้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

หากคนสองคนกำลังแทรกในเวลาเดียวกันตราบใดที่พวกเขากำลังใช้cursors ที่แตกต่างกัน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

เครดิตทั้งหมดให้กับ@Martijn Pietersในความคิดเห็น:

คุณสามารถใช้ฟังก์ชั่นlast_insert_rowid():

last_insert_rowid()ฟังก์ชันส่งกลับROWIDของแทรกแถวสุดท้ายจากการเชื่อมต่อฐานข้อมูลที่เรียกฟังก์ชั่น last_insert_rowid()ฟังก์ชั่น SQL เป็นเสื้อคลุมรอบsqlite3_last_insert_rowid()ฟังก์ชั่นอินเตอร์เฟซ C / C ++