remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-09-13 20:14:24 +02:00
Code Issues Projects Releases Packages Wiki Activity

upsert new rows with constraints

Browse source
This commit is contained in:
Colin Dellow 2022-11-26 11:11:06 -05:00
commit 32f8173a8f
2 changed files with 21 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
sqlite_utils/db.py
View file

@ -2779,13 +2779,16 @@ class Table(Queryable):
self.last_pk = None
for record_values in values:
# TODO: make more efficient:
# The initial insert statement includes all columns, so that upserts
# of new rows whose non-pk columns have constraints can succeed
record = dict(zip(all_columns, record_values))
sql = "INSERT OR IGNORE INTO [{table}]({pks}) VALUES({pk_placeholders});".format(
sql = "INSERT OR IGNORE INTO [{table}]({columns}) VALUES({column_placeholders});".format(
table=self.name,
pks=", ".join(["[{}]".format(p) for p in pks]),
pk_placeholders=", ".join(["?" for p in pks]),
columns=", ".join(["[{}]".format(p) for p in all_columns]),
column_placeholders=", ".join(["?" for p in all_columns]),
)
queries_and_params.append((sql, [record[col] for col in pks]))
queries_and_params.append((sql, [record[col] for col in all_columns]))
# UPDATE [book] SET [name] = 'Programming' WHERE [id] = 1001;
set_cols = [col for col in all_columns if col not in pks]
if set_cols: