如何在Python中使用上下文管理器定义默认值？能否实现示例代码逻辑？

要实现你想要的效果，核心是让上下文管理器B在生效期间，临时调整A类的实例化逻辑，使其自动使用指定的默认值。以下是两种可行的实现方案：

方案一：临时替换A的构造方法 #

直接修改A类的__init__方法，在上下文内自动填充默认参数，退出上下文时恢复原方法，避免影响外部逻辑：

class A:
    def __init__(self, attr):
        self.attr = attr

class B:
    def __init__(self, default_attr):
        self.default_attr = default_attr
        self.original_init = A.__init__  # 保存原构造方法

    def __enter__(self):
        # 定义新构造方法，自动传入默认值
        def new_init(self_inst):
            self.original_init(self_inst, self.default_attr)
        A.__init__ = new_init

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 恢复原构造方法
        A.__init__ = self.original_init

# 测试验证
default_attr = 5
with B(default_attr):
    a = A()

assert a.attr == default_attr

方案二：通过类属性存储默认值 #

让A类本身支持读取上下文提供的默认值，无需替换整个构造方法，更灵活保留原类的传参能力：

class A:
    _default_attr = None  # 类级别存储默认值

    def __init__(self, attr=None):
        self.attr = attr if attr is not None else self._default_attr
        if self.attr is None:
            raise ValueError("attr参数未提供且无默认值")

class B:
    def __init__(self, default_attr):
        self.default_attr = default_attr
        self.original_default = A._default_attr

    def __enter__(self):
        A._default_attr = self.default_attr

    def __exit__(self, exc_type, exc_val, exc_tb):
        A._default_attr = self.original_default

# 测试验证
default_attr = 5
with B(default_attr):
    a = A()

assert a.attr == default_attr

注意事项 #

• 方案一直接修改类方法，多线程环境下可能存在线程安全问题，需额外处理；
• 方案二更友好，A类在上下文外实例化时仍需传入attr参数，符合原类的设计逻辑；
• 两种方案都在__exit__方法中恢复了原状态，确保上下文外的代码不受影响。

内容的提问来源于stack exchange，提问作者Guillermo Soto Gómez