Hypothetically, if __enter__ could be a generator, with the with body running for each yielded value, that would give us the power to implement class Loop: def __enter__(self): self._do_continue = ...

Splitting code into smaller functions is often good practice, but not always. I often find it clearer to have the whole inline, so it can be seen at a glance, instead of lots of small functions used ...