Python跨模块下上下文管理器动态定义外部变量失败的解决方法

Problem Analysis #

拆分模块后出现NameError的核心原因：原始实现中globals()返回的是定义类A的src模块的全局命名空间，而非调用with A('foo')的测试文件的命名空间。因此foo被绑定到了src模块的全局字典里，测试文件的命名空间中根本不存在这个变量。

Solution Code #

实现上下文管理器类A（src/namespace_manager.py） #

import inspect

class A:
    def __init__(self, name):
        self.name = name
        self.namespace = {}
        # 获取调用方的全局命名空间（关键修复点）
        caller_frame = inspect.currentframe().f_back
        self.caller_globals = caller_frame.f_globals

    def __enter__(self):
        # 将实例绑定到调用方的全局命名空间
        self.caller_globals[self.name] = self
        return self  # 保留return不影响，只是不需要用as语法

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 若需在with块外部保留foo，注释下面的删除代码
        # if self.name in self.caller_globals:
        #     del self.caller_globals[self.name]
        pass

    def add(self, obj):
        # 用对象的__name__作为命名空间的键
        self.namespace[obj.__name__] = obj

    def __getattr__(self, attr):
        # 动态返回命名空间中的对象
        if attr in self.namespace:
            return self.namespace[attr]
        raise AttributeError(f"'{type(self).__name__}' has no attribute '{attr}'")

测试代码（test.py） #

from src.namespace_manager import A

with A('foo'):
    class B:
        def __init__(self):
            print("B instance initialized")
    def c():
        return "Hello from function c"
    
    foo.add(B)
    foo.add(c)

# 在with块外部正常访问
foo.B()  # 输出: B instance initialized
print(foo.c())  # 输出: Hello from function c

Key Explanations #

1. 绑定到调用方命名空间：
   通过inspect.currentframe().f_back获取调用A('foo')的栈帧，再通过f_globals拿到调用方的全局字典，确保foo被绑定到测试文件的命名空间中，而非src模块。

2. 动态属性访问：
   __getattr__特殊方法会在访问实例不存在的属性时触发，这里从self.namespace中查找对应的类或函数，实现foo.B、foo.c的直接调用。

3. 命名空间管理：
   add方法利用对象的__name__属性作为键，将类和函数存入实例的namespace字典，保证命名与定义时一致。

Notes #

• 如果希望with块结束后自动清理foo变量，取消__exit__中删除代码的注释即可，但此时with块外部无法再访问foo。
• inspect模块的栈帧操作在常规Python环境中稳定可靠，适合包开发场景。

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