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

Implemented .convert(..., where=, where_args=), refs #304

Browse source
This commit is contained in:
Simon Willison 2021-08-02 11:33:56 -07:00
commit 69c7da5ec9
3 changed files with 55 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
sqlite_utils/db.py
View file

@ -1750,7 +1750,7 @@ class Table(Queryable):
if output not in self.columns_dict:
self.add_column(output, output_type or "text")
todo_count = self.count * len(columns)
todo_count = self.count_where(where, where_args) * len(columns)
with progressbar(length=todo_count, silent=not show_progress) as bar:
def convert_value(v):
@ -1760,7 +1760,7 @@ class Table(Queryable):
return fn(v)
self.db.register_function(convert_value)
sql = "update [{table}] set {sets};".format(
sql = "update [{table}] set {sets}{where};".format(
table=self.name,
sets=", ".join(
[
@ -1770,9 +1770,10 @@ class Table(Queryable):
for column in columns
]
),
where=" where {}".format(where) if where is not None else "",
)
with self.db.conn:
self.db.execute(sql)
self.db.execute(sql, where_args or [])
if drop:
self.transform(drop=columns)
return self
@ -1793,7 +1794,9 @@ class Table(Queryable):
for row in self.rows_where(
select=", ".join(
"[{}]".format(column_name) for column_name in (pks + [column])
)
),
where=where,
where_args=where_args,
):
row_pk = tuple(row[pk] for pk in pks)
if len(row_pk) == 1: