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

Fix KeyError when passing an unrecognized string type to create_table or add_column

Browse source 
Passing a valid SQLite type string that is not in COLUMN_TYPE_MAPPING, such as
NUMERIC or ANY, raised KeyError instead of being used as-is. The fix falls back
to the raw string value when the type is a string not present in the mapping,
matching the documented behaviour: col_type accepts Python types or SQLite type
strings.
This commit is contained in:
ikatyal2110 2026-07-27 14:42:17 +00:00
commit 5d812b920f
No known key found for this signature in database
2 changed files with 25 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
sqlite_utils/db.py
View file

@ -1480,7 +1480,9 @@ class Database:
column_extras.append(
f"REFERENCES {quote_identifier(fk.other_table)}({quote_identifier(cast(str, fk.other_column))}){_fk_actions_sql(fk)}"
)
column_type_str = COLUMN_TYPE_MAPPING[column_type]
column_type_str = COLUMN_TYPE_MAPPING.get(column_type) or (
column_type if isinstance(column_type, str) else COLUMN_TYPE_MAPPING[column_type]
)
# Special case for strict tables to map FLOAT to REAL
# Refs https://github.com/simonw/sqlite-utils/issues/644
if strict and column_type_str == "FLOAT":
@ -3121,7 +3123,9 @@ class Table(Queryable):
sql = "ALTER TABLE {} ADD COLUMN {} {col_type}{not_null_default};".format(
quote_identifier(self.name),
quote_identifier(col_name),
col_type=fk_col_type or COLUMN_TYPE_MAPPING[col_type],
col_type=fk_col_type or COLUMN_TYPE_MAPPING.get(col_type) or (
col_type if isinstance(col_type, str) else COLUMN_TYPE_MAPPING[col_type]
),
not_null_default=(" " + not_null_sql) if not_null_sql else "",
)
self.db.execute(sql)