vassili/seitime-frappe
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

refactor(utils): Update create_batch docstring, annotations

Browse source
This commit is contained in:
Gavin D'souza 2021-02-04 17:21:06 +05:30
parent 3589f724af
commit ff27940b8e
1 changed files with 11 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
frappe/utils/__init__.py
View file

@ -700,13 +700,19 @@ def get_safe_filters(filters):
return filters
def create_batch(iterable, batch_size):
"""
Convert an iterable to multiple batches of constant size of batch_size
def create_batch(iterable: Iterable, size: int) -> Generator[Iterable, None, None]:
"""Convert an iterable to multiple batches of constant size of batch_size
Args:
iterable (Iterable): Iterable object which is subscriptable
size (int): Maximum size of batches to be generated
Yields:
Generator[List]: Batched iterable of maximum length `size`
"""
total_count = len(iterable)
for i in range(0, total_count, batch_size):
yield iterable[i:min(i + batch_size, total_count)]
for i in range(0, total_count, size):
yield iterable[i : min(i + size, total_count)]
def set_request(**kwargs):
from werkzeug.test import EnvironBuilder