Python：如何在上下文管理器结束后修改变量的值？

你想要实现这样的效果：

with my_context() as value:
    print(value)  # 上下文内输出1
print(value)      # 上下文结束后输出2

但你尝试的两种方法都没生效：

import contextlib

@contextlib.contextmanager
def number_function():
    number = 1
    yield number
    number = 2

class number_class:
    def __enter__(self):
        return 1
    def __exit__(self, exc_type, exc_value, traceback):
        return 2

with number_function() as number:
    print(number)  # 输出1
print(number)      # 还是输出1
with number_class() as number:
    print(number)  # 输出1
print(number)      # 还是输出1

为什么之前的方法无效？ #

• 对于contextlib.contextmanager装饰的生成器：你yield的是不可变整数1，生成器内部后续修改number = 2只是重新绑定了内部变量，不会影响外部通过as绑定的number——因为不可变类型传递的是值，不是引用。
• 对于类实现的上下文管理器：__exit__方法的返回值是用来控制是否抑制异常的（返回True会阻止异常向上传播），不是用来更新as绑定的变量的。

可行的实现方式 #

要让上下文结束后能修改外部可见的值，核心是传递可变对象——因为可变对象的修改会直接反映到引用它的地方，不需要重新绑定变量。

方法1：用可变容器（比如列表）简化实现

这是最简洁的方式之一，不需要自定义类：

import contextlib

@contextlib.contextmanager
def number_function():
    # 用列表包裹数值，列表是可变对象
    number = [1]
    yield number
    # 修改列表内的元素，外部能看到变化
    number[0] = 2

with number_function() as number:
    print(number[0])  # 输出1
print(number[0])      # 输出2

方法2：用简单的可变类（比如dataclass）

如果觉得列表的[0]不够直观，可以用dataclass定义一个极简的可变数值容器：

from dataclasses import dataclass
import contextlib

@dataclass
class MutableInt:
    value: int

@contextlib.contextmanager
def number_function():
    num = MutableInt(1)
    yield num
    num.value = 2

with number_function() as num:
    print(num.value)  # 输出1
print(num.value)      # 输出2

方法3：你提到的自定义上下文类

这是最直观的面向对象实现，虽然代码稍多，但可读性强：

class my_context:
    def __enter__(self):
        self.number = 1
        return self
    def __exit__(self, exc_type, exc_value, traceback):
        self.number = 2

with my_context() as context:
    print(context.number)  # 输出1
print(context.number)      # 输出2

总结 #

没有办法直接修改as绑定的不可变变量，但通过传递可变对象（容器、自定义类实例等），就能让上下文结束后的修改被外部感知。其中用列表或dataclass的方式，比你最初的自定义上下文类更简洁。

内容的提问来源于stack exchange，提问作者user3310334