1. ホーム
  2. パイソン

[解決済み】Pythonを使用してSQLiteに行を挿入した後、挿入されたIDを取得する方法は?

2022-04-11 23:25:25

質問

Pythonを使用してSQLiteに行を挿入した後、挿入されたidを取得するには?このようなテーブルがあります。

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

例のデータで新しい行を挿入します username="test"password="test" . 生成されたidをトランザクションセーフな方法で取得するには?これは、2人の人が同時にデータを挿入する可能性があるウェブサイトのソリューションのためです。私は最後に読まれた行を得ることができることを知っているが、私はそれがトランザクションセーフであるとは思わない。誰かが私にいくつかのアドバイスを与えることができますか?

どのように解決するのですか?

を使用することができます。 カーソル.lastrowid (参照 "Optional DB API Extensions"):

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人が同時に挿入する場合、それぞれ別の cursor s, 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

なお lastrowidNone で一度に複数行を挿入した場合。 executemany :

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