使用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:
                ...