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

Show all missing columns when raising the missing error.

Browse source
This commit is contained in:
Ethan Dalool 2020-02-06 20:17:49 -08:00
parent 8238481262
commit 6cac1d5a25
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
voussoirkit/sqlhelpers.py
View file

@ -48,13 +48,16 @@ def insert_filler(column_names, values, require_all=True):
cur.execute(query, bindings) cur.execute(query, bindings)
''' '''
values = values.copy() values = values.copy()
missings = []
for column in column_names: for column in column_names:
if column in values: if column in values:
continue continue
if require_all: if require_all:
raise ValueError('Missing column "%s"' % column) missings.append(column)
else: else:
values[column] = None values[column] = None
if missings:
raise ValueError(f'Missing columns {missings}.')
qmarks = '?' * len(column_names) qmarks = '?' * len(column_names)
qmarks = ', '.join(qmarks) qmarks = ', '.join(qmarks)
bindings = [values[column] for column in column_names] bindings = [values[column] for column in column_names]