Question

我想从已安装的模块上对f(*args, **kwargs)进行修补。我在自己的代码上使用了decorator的想法，但是已安装模块中的其他方法无法正确调用f。

这里是一个例子：

import numpy as np

def log(func):
    def wrapper(*args, **kwargs):
        print('logging')
        return func(*args, **kwargs)
    return wrapper

if __name__ == "__main__":
    a1 = np.asarray([0, 1, 2])
    print(f'a1={a1}')

    a2 = np.array([0, 1, 2])
    print(f'a2={a2}')

    np.array = log(np.array)

    a3 = np.asarray([0, 1, 2])
    print(f'a3={a3}')

    a4 = np.array([0, 1, 2])
    print(f'a4={a4}')

输出为：

a1=[0 1 2]
a2=[0 1 2]
a3=[0 1 2]
logging
a4=[0 1 2]

我希望结果是：

a1=[0 1 2]
a2=[0 1 2]
logging
a3=[0 1 2]
logging
a4=[0 1 2]

因为asarray在源代码中调用了array。

我的问题是： 1.为什么猴子补丁失败？ 2.如何解决？

Answer 1

通过np.array = log(np.array)，您正在更改“公共”名称np.array所指的功能。

但是np.asarray是在numpy.core.numeric模块中定义的，该模块具有自己的“私有”名称来引用该功能。不受公共名称修补的影响。

您必须修补私人名称：

np.core.numeric.array = log(np.array)

Python-猴子补丁失败，为什么？

1 个答案: