如何修改基于类的上下文管理器的对象？

Question

我想在with 范围内修改上下文管理器的对象。考虑以下代码：

class CM:

    def __init__(self):
        self.a = 1

    def __enter__(self):
        self.a = 2
        return self.a

    def __exit__(self, exc_type, exc_val, exc_tb):
        print(self.a)


with CM() as a:
    a = 3
    

我想在with 范围内将3 分配给self.a（因此看到3 的输出）。代码输出2，即类中self.a的值。

是否可以在类之外修改它？（类比是修改对象属性）

Answer 1

找到了，我需要返回对象本身：

class CM:

    def __init__(self):
        self.a = 1

    def __enter__(self):
        self.a = 2
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        print(self.a)


with CM() as a:
    a.a = 3