example-code-2e/17-it-generator/yield_delegate_fix.py
2019-04-06 07:11:08 -03:00

25 lines
460 B
Python

""" Example adapted from ``yield_delegate_fail.py``
The following program performs a simple abstraction over the process of
yielding.
"""
# BEGIN YIELD_DELEGATE_FIX
def f():
def do_yield(n):
yield n
x = 0
while True:
x += 1
yield from do_yield(x)
# END YIELD_DELEGATE_FIX
if __name__ == '__main__':
print('Invoking f() now produces a generator')
g = f()
print(next(g))
print(next(g))
print(next(g))