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

问题描述

我想在 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的值。

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

解决方法

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

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