Inheritance with contextlib.contextmanager

Question

Given a class

class SomeClass:
    @contextlib.contextmanager
    def on_connection(self, target_terminal, source_terminal):
        ...
        yield
        ...

How do you inherit from it?

Answer 1

It's possible to combine the contextlib.contextmanager pattern with a with block to bring in the superclass' context manager:

class SomeDerivedClass(SomeClass):
    @contextlib.contextmanager
    def on_connection(self, target_terminal, source_terminal):
        with super().on_connection(target_terminal, source_terminal):
            ...
            try:
                yield
            finally:
                ...