我们可以将ContextManager Decorator与__Enter __（）和__EXIT __（）方法混合使用，而__（）方法与语句中的另一个类中的其他类中的方法相混合？

Question

在python3.8中，我非常熟悉传统__输入__和__退出__魔术方法，但@coce>@contextlib.contlib.contextmanager decorator。是否可以将单个中的两个模式与语句混合在一起？

以下（高度人为）脚本应该更清楚地解释问题。 contextClass.enter_context_function（）和contextClass.exit_context_function（）是否有定义仅使用context_function（）函数并使单元测试通过？还是这些模式是互斥的？

import contextlib


NUMBERS = []


@contextlib.contextmanager
def context_function():
    NUMBERS.append(3)
    try:
        yield
    finally:
        NUMBERS.append(5)


class ContextClass:
    def __init__(self):
        self.numbers = NUMBERS
        self.numbers.append(1)

    def __enter__(self):
        self.numbers.append(2)
        self.enter_context_function() # should append 3
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.exit_context_function() # should append 5
        self.numbers.append(6)

    def function_call(self):
        self.numbers.append(4)

    def enter_context_function(self):
        # FIX ME!
        pass

    def exit_context_function(self):
        # FIX ME!
        pass


if __name__ == "__main__":
    import unittest

    class TestContextManagerFunctionAndClass(unittest.TestCase):
        def test_context_function_and_class(self):
            with ContextClass() as cc:
                cc.function_call()
            self.assertEqual(NUMBERS, [1, 2, 3, 4, 5, 6])

    unittest.main()

我知道有更好的方法可以解决一个类似的问题（特别是重写context_function作为具有其自己_____ __和__ exit> __ exit __方法的类，但是我M试图更好地了解上下文Manager Decorator的工作方式。

Answer 1

__ Init __无需更改。 “使单位测试通过”的手册方式是：

def enter_context_function(self):
    self._context_mgr = context_function()
    self._context_mgr.__enter__()

def exit_context_function(self):
    self._context_mgr.__exit__(None, None, None)

但是，有点缺少上下文管理者的点。它们旨在在a

另请注意，如书面，如果加剧后代码，则无法达到 line（5） line（“拆卸”）。应该像这样写：

@contextlib.contextmanager
def context_function():
    NUMBERS.append(3)
    try:
        yield
    finally:
        NUMBERS.append(5)