remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-07-26 19:04:32 +02:00
Code Issues Projects Releases Packages Wiki Activity

Tests for analyze tables

Browse source
This commit is contained in:
Simon Willison 2020-12-12 20:46:13 -08:00
commit f7efe77ccb
2 changed files with 103 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

15
sqlite_utils/cli.py
View file

@ -1361,7 +1361,14 @@ def insert_files(
required=True,
)
@click.argument("tables", nargs=-1)
@click.option("-c", "--column", "columns", type=str, multiple=True, help="Specific columns to analyze")
@click.option(
"-c",
"--column",
"columns",
type=str,
multiple=True,
help="Specific columns to analyze",
)
@click.option("--save", is_flag=True, help="Save results to _analyze_tables table")
@load_extension_option
def analyze_tables(
@ -1381,11 +1388,13 @@ def analyze_tables(
for table in tables:
table_counts[table] = db[table].count
for column in db[table].columns:
if (not columns or column.name in columns):
if not columns or column.name in columns:
todo.append((table, column.name))
# Now we now how many we need to do
for i, (table, column) in enumerate(todo):
column_details = db[table].analyze_column(column, total_rows=table_counts[table])
column_details = db[table].analyze_column(
column, total_rows=table_counts[table]
)
if save:
db["_analyze_tables"].insert(
column_details._asdict(), pk=("table", "column"), replace=True