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

fix: add input validation to chunks() for size < 1

Browse source
This commit is contained in:
KooshaPari 2026-09-12 23:13:52 -07:00
commit d5b4235968
2 changed files with 12 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
sqlite_utils/utils.py
View file

@ -577,6 +577,8 @@ def chunks(sequence: Iterable[T], size: int) -> Iterable[Iterable[T]]:
:param sequence: Any Python iterator
:param size: The size of each chunk
"""
if size < 1:
raise ValueError(f"chunk size must be at least 1, got {size}")
iterator = iter(sequence)
for item in iterator:
yield itertools.chain([item], itertools.islice(iterator, size - 1))