python - contextlib.contextmanager による継承

Question

与えられたクラス

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

そこからどのように継承しますか？

Answer 1

contextlib.contextmanagerパターンをブロックと組み合わせてwith、スーパークラスのコンテキスト マネージャーを取り込むことができます。

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