python - 修补本地实例

Question

考虑我在 python 模块 a.py 中有以下代码:

def func():
    obj = Obj()
    # Some code that uses obj

有没有办法在不将 obj 设为函数参数的情况下对其进行猴子修补？

原因:在代码本身的上下文中，它不应该是一个参数，但是在使用代码进行测试和系统模拟的上下文中，补丁是合理的。此外，在各种情况下，我可能需要修补不同的本地对象，并将它们全部作为参数放置将是乏味的。

Answer 1

# if func() is defined in `a.py`
import a

oldObj, a.Obj = a.Obj, MockObj
a.func()
a.Obj = oldObj