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

Support ANY column types for strict tables

Browse source 
Closes #790, #820
This commit is contained in:
Simon Willison 2026-08-12 16:43:33 -07:00
commit fcfccea813
13 changed files with 292 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

14
sqlite_utils/db.py
View file

@ -39,6 +39,7 @@ from .create_table_parser import (
sql_ends_in_line_comment,
)
from .utils import (
ANY,
OperationalError,
chunks,
column_affinity,
@ -366,6 +367,7 @@ COLUMN_TYPE_MAPPING: dict[Any, str] = {
decimal.Decimal: "REAL",
None.__class__: "TEXT",
uuid.UUID: "TEXT",
ANY: "ANY",
# SQLite explicit types
"TEXT": "TEXT",
"INTEGER": "INTEGER",
@ -380,6 +382,8 @@ COLUMN_TYPE_MAPPING: dict[Any, str] = {
"real": "REAL",
"blob": "BLOB",
"bytes": "BLOB",
"ANY": "ANY",
"any": "ANY",
}
# If numpy is available, add more types
if np:
@ -3092,6 +3096,15 @@ class Table(Queryable):
if col in columns
}
if lookup_table.exists():
if (
self.strict
and ANY in lookup_columns_definition.values()
and not lookup_table.strict
):
raise InvalidColumns(
f"Lookup table {table} already exists but is not STRICT, "
"so it cannot preserve ANY column values"
)
if not set(lookup_columns_definition.items()).issubset(
lookup_table.columns_dict.items()
):
@ -3105,6 +3118,7 @@ class Table(Queryable):
**lookup_columns_definition,
},
pk="id",
strict=self.strict,
)
lookup_columns = [(rename.get(col) or col) for col in columns]
lookup_table.create_index(lookup_columns, unique=True, if_not_exists=True)