Please see the below snippet, run with Python 3.10:
from collections.abc import Generator
DUMP_DATA = 5, 6, 7
class DumpData(Exception):
"""Exception used to indicate to yield from DUMP_DATA."""
def sample_gen() -> Generator[int | None, int, None]:
out_value: int | None = None
while True:
try:
in_value = yield out_value
except DumpData:
yield len(DUMP_DATA)
yield from DUMP_DATA
out_value = None
continue
out_value = in_value
My question pertains to the DumpData
path where there is a yield from
. After that yield from
, there needs to be a next(g)
call, to bring the generator
back to the main yield
statement so we can send
:
def main() -> None:
g = sample_gen()
next(g) # Initialize
assert g.send(1) == 1
assert g.send(2) == 2
# Okay let's dump the data
num_data = g.throw(DumpData)
data = tuple(next(g) for _ in range(num_data))
assert data == DUMP_DATA
# How can one avoid this `next` call, before it works again?
next(g)
assert g.send(3) == 3
How can this extra next
call be avoided?
from Avoiding extra `next` call after `yield from` in Python generator
No comments:
Post a Comment