alternatively, the

@contextmanager

annotation seems to make this much, much easier.