a context manager / iterator which remembers the final value

context_manager_iterator.py

import random 

class Foo():

    data = (random.randint(0, 1000) for x in range(10))
    last = None

    def __iter__(self):
        return self

    def __next__(self):
        self.last = next(self.data)
        return self.last

    def __enter__(self):
        return self

    def __exit__(self, _e, _v, _t):
        pass

with Foo() as f:
    for x in f:
        print(x)
    print("final value {}".format(f.last))