Jak odzyskać wstawiony identyfikator po wstawieniu wiersza w SQLite za pomocą Pythona?

Question

Jak odzyskać wstawiony identyfikator po wstawieniu wiersza w SQLite za pomocą Pythona?

Jak odzyskać wstawione id po wstawieniu wiersza w SQLite za pomocą Pythona? Mam taką tabelkę:

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

Wstawiam nowy wiersz z przykładowymi danymi username="test" i password="test". Jak odzyskać wygenerowany identyfikator w bezpieczny sposób? Jest to rozwiązanie internetowe, w którym dwie osoby mogą wstawiać dane w tym samym czasie. Wiem, że mogę dostać ostatni wiersz, ale nie sądzę, że jest to bezpieczne transakcja. Czy ktoś może mi coś doradzić?

127

python sqlite sqlite3

Author: gustafbstrom, 2011-06-05

Source

1 answers

score 193 · Accepted Answer

Możesz użyć kursora.lastrowid (patrz "opcjonalne rozszerzenia API DB"):

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

Jeśli dwie osoby wstawiają w tym samym czasie, o ile używają różnych cursors, cursor.lastrowid zwróci id dla ostatniego wiersza, który cursor wstawiono:

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

Zauważ, że lastrowid zwraca None gdy wstawisz więcej niż jeden wiersz na raz z executemany:

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