Как правильно записать данные в две таблицы БД SQLite?

Question

[pypyshka]

Добрый день!

Подскажите, пожалуйста, как правильно сделать запросы в данном случае. Есть два списка:
a = [('1', '2', '3')]
b = [('a', 'b', 'c'), ('x', 'y', 'z')]

В БД есть две таблицы:
table_a:
id | value1 | value2 | value 3 - здесь id - первичный ключ с автоинкрементом

table_b:
id | value1 | value2 | value 3

Соответственно делаю запросы:

query_a = "INSERT INTO table_a (value1, value2, value 3) VALUES (?, ?, ?)"
query_b = "INSERT INTO table_b (value1, value2, value 3) VALUES (?, ?, ?)"
cur.executemany(query_a, a)
cur.executemany(query_b, b)

и получаю:
table_a:
id | value1 | value2 | value 3
1 | 1 | 2| 3

table_b:
id | value1 | value2 | value 3
- | a | b | c
- | x | y | z

А как правильно сделать, чтобы в table_b в поле id бралось значение id из table_a, то есть чтобы получить:
table_b:
id | value1 | value2 | value 3
1 | a | b | c
1 | x | y | z

Comments

[DDDsa]

Во вторую таблицу нужно вставить именно 1? Как определить, какой ID нужно подставить во вторую таблицу?

[pypyshka]

Даниил: сначала мы выполняем первый запрос: вставляем данные в table_a, в этой таблице автоматически задается значение для поля id (в данном случае id = 1, при следующей вставке будет 2 и т. д.). После того как данные записались в table_a, мы вставляем данные в table_b. Но как теперь сделать, чтобы id из table_a, который на данный момент равен 1, так же был указан и в table_b в поле id?

Answer 1

[DDDsa]

Если я правильно понял задачу,

query_a = "INSERT INTO table_a (value1, value2, value 3) VALUES (?, ?, ?)"
cur.execute(query_a, a[0])
last_id = cur.lastrowid

теперь значение переменной last_id можно подставлять во второй инзерт
Или так:

query_a = "INSERT INTO table_a (value1, value2, value 3) VALUES (?, ?, ?)"
cur.executemany(query_a, a)
cur.execute('SELECT last_insert_rowid()')
last_id = cur.fetchone()[0]

Comments

[pypyshka]

Почему-то last_id возвращает None

[DDDsa]

pypyshka: Вы правы, cur.lastrowid возвращает None после executemany. Работает только после execute.
Тогда другой вариант:

cur.execute('SELECT last_insert_rowid()')
 last_id = cur.fetchone()[0]

[pypyshka]

Даниил: этот вариант также возвращает ничего

[DDDsa]

pypyshka: Ну как же, вот, только что прогнал:

>>> import sqlite3
>>> c = sqlite3.connect(r'c:\test.sqlite')
>>> cur = c.cursor()
>>> cur.execute('create table test1(id number auto increment primary key, f1 text, f2 text, f3 text)')
<sqlite3.Cursor object at 0x006961A0>
>>> cur.execute('create table test2(id number, f1 text, f2 text, f3 text)')
<sqlite3.Cursor object at 0x006961A0>
>>> query_a = "INSERT INTO test1 (f1, f2, f3) VALUES (?, ?, ?)"
>>> a = [('1', '2', '3')]
>>> cur.executemany(query_a,a)
<sqlite3.Cursor object at 0x006961A0>
>>> cur.execute('select last_insert_rowid()')
<sqlite3.Cursor object at 0x006961A0>
>>> cur.fetchone()
(1,)

[pypyshka]

Даниил: прошу прощения, это я был невнимателен. Действительно, все работает. Огромное вам спасибо за помощь!