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

insert now replaces square braces in column name with underscore, closes #341

Browse source
This commit is contained in:
Simon Willison 2021-11-14 18:56:35 -08:00
commit ffb54427d3
2 changed files with 14 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

14
sqlite_utils/db.py
View file

@ -2571,6 +2571,8 @@ class Table(Queryable):
all_columns = []
first = True
num_records_processed = 0
# Fix up any records with square braces in the column names
records = fix_square_braces(records)
# We can only handle a max of 999 variables in a SQL insert, so
# we need to adjust the batch_size down if we have too many cols
records = iter(records)
@ -2618,7 +2620,6 @@ class Table(Queryable):
column for column in record if column not in all_columns
]
validate_column_names(all_columns)
first = False
self.insert_chunk(
@ -2986,3 +2987,14 @@ def validate_column_names(columns):
assert (
"[" not in column and "]" not in column
), "'[' and ']' cannot be used in column names"
def fix_square_braces(records: Iterable[Dict[str, Any]]):
for record in records:
if any("[" in key or "]" in key for key in record.keys()):
yield {
key.replace("[", "_").replace("]", "_"): value
for key, value in record.items()
}
else:
yield record