remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-07-27 11:24: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/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))