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

Add --extract option to insert/upsert commands

Browse source 
Wires the extracts= Python API feature through the CLI so imports can
create lookup tables in one step: --extract species for a table named
after the column, or --extract species:Species for a custom name.

Closes #352
This commit is contained in:
Parker Gurney 2026-08-05 14:22:45 -07:00
commit d4d075c924
5 changed files with 111 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

17
sqlite_utils/cli.py
View file

@ -1026,6 +1026,14 @@ def insert_upsert_options(*, require_pk=False):
type=(str, str),
help="Default value that should be set for a column",
),
click.option(
"--extract",
"extract",
multiple=True,
help="Extract this column into a separate lookup table, e.g. "
"--extract species or --extract species:Species to use a "
"custom table name",
),
click.option(
"--type",
"types",
@ -1091,6 +1099,7 @@ def insert_upsert_implementation(
truncate=False,
not_null=None,
default=None,
extract=None,
types=None,
no_detect_types=False,
analyze=False,
@ -1119,6 +1128,10 @@ def insert_upsert_implementation(
extra_kwargs["not_null"] = set(not_null)
if default:
extra_kwargs["defaults"] = dict(default)
if extract:
extra_kwargs["extracts"] = {
item.split(":", 1)[0]: item.split(":", 1)[-1] for item in extract
}
if column_type_overrides:
extra_kwargs["columns"] = column_type_overrides
if upsert:
@ -1405,6 +1418,7 @@ def insert(
truncate,
not_null,
default,
extract,
types,
strict,
):
@ -1500,6 +1514,7 @@ def insert(
silent=silent,
not_null=not_null,
default=default,
extract=extract,
types=types,
strict=strict,
code=code,
@ -1536,6 +1551,7 @@ def upsert(
alter,
not_null,
default,
extract,
types,
no_detect_types,
analyze,
@ -1588,6 +1604,7 @@ def upsert(
key=key,
not_null=not_null,
default=default,
extract=extract,
types=types,
no_detect_types=no_detect_types,
analyze=analyze,