Python如何注入代码到模块globals以绕过模块导入？

看来你想通过模块注入的方式，让用户导入mymodule后就能直接调用mymodule.foo，但直接定义foo的方式满足不了需求，尝试修改模块globals字典触发注入还报了错，同时也考虑过预加载函数的思路。我来给你梳理几个靠谱的解决方案：

1. 用模块级__getattr__实现懒加载注入（推荐） #

Python 3.7+支持模块级的__getattr__魔法方法，当访问模块中不存在的属性时，会自动调用这个函数。这刚好能解决你“找不到foo时触发注入逻辑”的需求，而且比直接修改globals字典更稳妥，不会有作用域或时机问题。

在mymodule/__init__.py里写这段代码：

def __getattr__(name):
    if name == 'foo':
        # 这里写你的注入逻辑：比如动态导入子模块的foo，或者动态生成函数
        def foo():
            print("这是动态注入的foo函数！")
        
        # 把生成的foo存入模块globals，避免每次调用都重新生成
        globals()['foo'] = foo
        return foo
    # 其他属性找不到时抛出标准错误
    raise AttributeError(f"module '{__name__}' has no attribute '{name}'")

这样，当用户第一次调用mymodule.foo()时，会触发__getattr__执行注入逻辑，之后再调用就直接用缓存到globals里的foo了，完全符合你的需求。

2. 预加载函数的具体实现 #

如果你的场景允许预加载（不需要懒加载，或者foo的生成逻辑不需要依赖导入时的上下文），直接在__init__.py里提前初始化foo即可：

情况1：foo来自子模块 #

# 从子模块导入foo，用户导入mymodule后直接能用
from .internal_module import foo

情况2：动态生成foo后预加载 #

def _generate_foo():
    # 这里写生成foo的自定义逻辑
    def foo():
        return "预加载的foo函数"
    return foo

# 初始化foo并存入模块命名空间
foo = _generate_foo()

关于你修改globals字典报错的可能原因 #

你之前尝试修改模块globals字典触发注入时报错，大概率是因为时机不对——比如在模块初始化阶段修改globals，但属性查找逻辑已经走过了；或者没有正确处理属性查找的触发条件，导致注入逻辑没在合适的时机执行。而__getattr__是Python官方设计来处理这类动态属性场景的，能避开这些坑。

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