voussoir/voussoirkit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add parameter ignore_duplicates to insert function.

Browse source
This commit is contained in:
voussoir 2025-03-24 22:15:43 -07:00
parent d9a5d0811d
commit 9385b365cd
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
voussoirkit/worms.py
View file

@ -433,12 +433,15 @@ class Database(metaclass=abc.ABCMeta):
tables = set(self.select_column(query)) tables = set(self.select_column(query))
return tables return tables
def insert(self, table, pairs) -> sqlite3.Cursor: def insert(self, table, pairs, *, ignore_duplicate=False) -> sqlite3.Cursor:
if isinstance(table, type) and issubclass(table, Object): if isinstance(table, type) and issubclass(table, Object):
table = table.table table = table.table
self.assert_table_exists(table) self.assert_table_exists(table)
(qmarks, bindings) = sqlhelpers.insert_filler(pairs) (qmarks, bindings) = sqlhelpers.insert_filler(pairs)
query = f'INSERT INTO {table} {qmarks}' if ignore_duplicate:
query = f'INSERT OR IGNORE INTO {table} {qmarks}'
else:
query = f'INSERT INTO {table} {qmarks}'
return self.execute(query, bindings) return self.execute(query, bindings)
def normalize_object_id(self, object_class, object_id): def normalize_object_id(self, object_class, object_id):