Though just tossing that onto a method with

@contextmanager

also seems just as good, so