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

Progress bar for "sqlite-utils extract", closes #169

Browse source
This commit is contained in:
Simon Willison 2020-09-22 17:02:29 -07:00
commit 5c4d58d152
3 changed files with 26 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
sqlite_utils/db.py
View file

@ -883,7 +883,7 @@ class Table(Queryable):
return sqls
def extract(self, columns, table=None, fk_column=None, rename=None):
def extract(self, columns, table=None, fk_column=None, rename=None, progress=None):
rename = rename or {}
if isinstance(columns, str):
columns = [columns]
@ -905,6 +905,8 @@ class Table(Queryable):
row_pks = tuple(row[pk] for pk in pks)
lookups = {rename.get(column) or column: row[column] for column in columns}
self.update(row_pks, {first_column: lookup_table.lookup(lookups)})
if progress:
progress(1)
fk_column = fk_column or "{}_id".format(table)
# Now rename first_column and change its type to integer, and drop
# any other extracted columns: