python - 使用 contextlib.contextmanager 继承

Question

给定一个类

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

你如何继承它？

Answer 1

可以将 contextlib.contextmanager 模式与 with block 结合起来以引入父类(super class)的上下文管理器:

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