Fun with Generators

genfun.py

"""A hack to play around with generator acknowledgement."""


def extract():
    for i in xrange(10):
        ack = yield i
        print('acking %d' % ack)


def transform(ext):
    ack = None
    while True:
        message = ext.send(ack)
        ack = yield message**2


def load(xfmr):
    ack = None
    while True:
        try:
            message = xfmr.send(ack)
            ack = message
        except StopIteration:
            print('all finished')
            break


def main():
    load(transform(extract()))


if __name__ == '__main__':
    main()