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

Add test for chunks(), refs #364

Browse source
This commit is contained in:
Simon Willison 2022-01-08 13:16:34 -08:00
commit a8f9cc6f64
3 changed files with 23 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

7
sqlite_utils/db.py
View file

@ -1,4 +1,5 @@
from .utils import (
chunks,
sqlite3,
OperationalError,
suggest_column_types,
@ -2995,12 +2996,6 @@ class View(Queryable):
)
def chunks(sequence, size):
iterator = iter(sequence)
for item in iterator:
yield itertools.chain([item], itertools.islice(iterator, size - 1))
def jsonify_if_needed(value):
if isinstance(value, decimal.Decimal):
return float(value)

7
sqlite_utils/utils.py
View file

@ -3,6 +3,7 @@ import contextlib
import csv
import enum
import io
import itertools
import json
import os
from . import recipes
@ -306,3 +307,9 @@ def _compile_code(code, imports, variable="value"):
globals[import_.split(".")[0]] = __import__(import_)
exec(code_o, globals, locals)
return locals["fn"]
def chunks(sequence, size):
iterator = iter(sequence)
for item in iterator:
yield itertools.chain([item], itertools.islice(iterator, size - 1))