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

176

Jak odzyskać wstawiony identyfikator po wstawieniu wiersza w SQLite za pomocą Pythona? Mam taki stół:

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? Dotyczy to rozwiązania internetowego, w którym dwie osoby mogą wprowadzać dane w tym samym czasie. Wiem, że mogę uzyskać ostatni odczytany wiersz, ale nie sądzę, aby transakcja była bezpieczna. Czy ktoś może mi doradzić?

Jane
źródło

Odpowiedzi:

256

Możesz użyć cursor.lastrowid (zobacz „Opcjonalne rozszerzenia 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

Jeśli dwie osoby wstawiają w tym samym czasie, o ile używają różnych cursors, cursor.lastrowidzwróci wartość iddla ostatniego cursorwstawionego wiersza :

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

Zwróć uwagę, że lastrowidpowraca, Nonegdy wstawisz więcej niż jeden wiersz naraz z executemany:

cursor.executemany('INSERT INTO foo (username,password) VALUES (?,?)',
               (('baz','bar'),('bing','bop')))
print(cursor.lastrowid)
# None
unutbu
źródło
16
+1 za wyjaśnienie, że zwróci najnowszy identyfikator przez tę indywidualną instancję kursora.
quakkels
43
Dostępna jest również last_insert_rowid()funkcja SQL , która umożliwia wstawienie ostatniego identyfikatora wiersza jako klucza obcego w następnej instrukcji wstawiania, całkowicie w języku SQL.
Martijn Pieters
2
@MartijnPieters Możesz zamieścić to jako odpowiedź, mimo że pytanie jest dość stare. Wciąż może pomóc użytkownikom w czytaniu tej strony.
Daniel Werner
3
pojawia się Python sqlite3 lastrowid używa last_insert_rowidpod github.com/ghaering/pysqlite/blob/… (co nie jest gwarantowane dla wątków, ale wydaje się jedyną opcją FWIW). Zobacz także stackoverflow.com/q/2127138/32453
rogerdpack
2
Przy okazji, to działa, INSERTale nie działa UPDATE. Ale podczas aktualizowania wiersza prawdopodobnie i tak masz już odpowiedni identyfikator wiersza (po prostu zajęło mi to trochę czasu, zanim to zrozumiałem ...).
CGFoX,
4

Wszystkie kredyty dla @Martijn Pieters w komentarzach:

Możesz użyć funkcji last_insert_rowid():

last_insert_rowid()Funkcja zwraca ROWIDostatniego wiersza z wkładką połączenia z bazą danych, która wywołała funkcję. Funkcja last_insert_rowid()SQL jest otoką wokół sqlite3_last_insert_rowid()funkcji interfejsu C / C ++.

Stefan van den Akker
źródło