web-dev-qa-db-ja.com

Pythonを使用してSQLiteに行を挿入した後に挿入されたIDを取得する方法は？

Pythonを使用してSQLiteに行を挿入した後に挿入されたIDを取得する方法は？このようなテーブルがあります：

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

サンプルデータusername="test"およびpassword="test"を含む新しい行を挿入します。生成されたIDをトランザクションセーフな方法で取得するにはどうすればよいですか？これは、2人が同時にデータを挿入する可能性のあるWebサイトソリューション用です。最後の読み取り行を取得できることは知っていますが、トランザクションが安全であるとは思いません。誰かアドバイスをいただけますか？

pythonsqlite
148
Jane

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

2人が同時に挿入している場合、異なる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
224
unutbu