如何在不修改外部代码库的情况下对Python函数做Monkey Patch？

核心原理 #

Python模块是单例对象，一旦导入就会缓存到sys.modules中。要让所有对funA的调用生效，必须在任何模块导入funA之前，修改subA模块对象中的funA引用；如果依赖模块（比如subB）已经提前导入，需要清除其缓存后重新导入，才能获取更新后的函数引用。

具体操作步骤 #

1. 确保子项目路径正确加入sys.path #

要让Python识别subonemodule，需将subProjectsDir/subProject目录加入sys.path。可以直接在myA.py中处理，避免依赖__init__.py的配置：

import sys
from pathlib import Path

# 计算子项目绝对路径
sub_project_root = Path(__file__).parent.parent / "subProjectsDir" / "subProject"
sys.path.append(str(sub_project_root.resolve()))

2. 提前导入并Patch目标函数 #

先导入subA模块，替换其中的funA为自定义函数：

# 导入要Patch的模块
from subonemodule.subtwomodule import subA

# 定义自定义替代函数
def myFunA(*args, **kwargs):
    # 这里编写你的自定义逻辑
    print("执行patched版本的funA")
    # 可选：如果需要保留原函数功能，可调用原函数
    # return subA.funA(*args, **kwargs)

# 替换模块中的原函数
subA.funA = myFunA

3. 清除缓存并重新导入依赖模块 #

如果subB已经被导入过，其内部缓存的funA还是原函数引用，需删除sys.modules中的缓存后重新导入：

# 清除subB模块的缓存（如果存在）
sub_b_module_name = "subonemodule.subtwomodule.subthreemodule.subB"
if sub_b_module_name in sys.modules:
    del sys.modules[sub_b_module_name]

# 重新导入subB
from subonemodule.subtwomodule.subthreemodule import subB

4. 测试Patch效果 #

调用subB.funB()，此时会执行你自定义的myFunA：

subB.funB()

常见问题排查 #

• Patch无效？ 检查导入顺序：必须先PatchsubA，再导入subB或其他引用funA的模块。
• 路径错误？ 打印sys.path确认subProject目录已被加入，或直接使用绝对路径避免相对路径问题。
• 模块缓存未清除？ 如果之前运行过代码，sys.modules中可能残留旧模块对象，需清除后重新导入。

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