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

Improved code compilation pattern, closes #472

Browse source
This commit is contained in:
Simon Willison 2022-08-26 22:20:09 -07:00
commit a46a5e3a9e
3 changed files with 24 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

9
sqlite_utils/utils.py
View file

@ -432,12 +432,11 @@ def progressbar(*args, **kwargs):
def _compile_code(code, imports, variable="value"):
locals = {}
globals = {"r": recipes, "recipes": recipes}
# If user defined a convert() function, return that
try:
exec(code, globals, locals)
return locals["convert"]
exec(code, globals)
return globals["convert"]
except (AttributeError, SyntaxError, NameError, KeyError, TypeError):
pass
@ -464,8 +463,8 @@ def _compile_code(code, imports, variable="value"):
for import_ in imports:
globals[import_.split(".")[0]] = __import__(import_)
exec(code_o, globals, locals)
return locals["fn"]
exec(code_o, globals)
return globals["fn"]
def chunks(sequence: Iterable, size: int) -> Iterable[Iterable]: