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

Applied Black, refs #526, #525

Browse source
This commit is contained in:
Simon Willison 2023-05-08 14:54:24 -07:00
commit fca3ef8cf2
3 changed files with 12 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

12
sqlite_utils/db.py
View file

@ -414,7 +414,9 @@ class Database:
fn_name = name or fn.__name__
arity = len(inspect.signature(fn).parameters)
if not replace and (fn_name, arity) in self._registered_functions:
raise FunctionAlreadyRegistered(f'Already registered function with name "{fn_name}" and identical arity')
raise FunctionAlreadyRegistered(
f'Already registered function with name "{fn_name}" and identical arity'
)
kwargs = {}
registered = False
if deterministic:
@ -2693,15 +2695,17 @@ class Table(Queryable):
return jsonify_if_needed(fn(v))
fn_name = fn.__name__
if fn_name == '<lambda>':
fn_name = f'lambda_{hash(fn)}'
if fn_name == "<lambda>":
fn_name = f"lambda_{hash(fn)}"
self.db.register_function(convert_value, name=fn_name)
sql = "update [{table}] set {sets}{where};".format(
table=self.name,
sets=", ".join(
[
"[{output_column}] = {fn_name}([{column}])".format(
output_column=output or column, column=column, fn_name=fn_name
output_column=output or column,
column=column,
fn_name=fn_name,
)
for column in columns
]