python能否修改import的模块的指针，从而实现重写方法的效果？

Question

python能否修改import的模块的指针，从而实现重写方法的效果？

我希望实现的效果是import一个a方法之后，使用另一个b方法替换掉a方法，之后任何地方import a 都相当于import b，类似于实现以下效果：
test_a.py

def method_a():
    print("aaa")

test_b.py

def method_b():
    print("bbb")

main.py

from test_a import method_a
method_a()

输出： bbb
我尝试使用 method_a = method_b的方式将method_a()指向method_b()，但是修改完的当时指针变了，其他地方使用method_a()指针就变回去了。

Answer 1

在Python中，可以通过修改导入的模块对象来实现重写方法的效果，但这种做法并不推荐，因为可能会引起意料之外的问题。

具体来说，Python在导入模块时会执行以下步骤：

搜索模块的路径，找到相应的.py文件；
编译模块的源代码，生成一个模块对象；
将模块对象添加到sys.modules字典中，以便下次导入时可以直接返回已经加载的模块对象。
在Python中，import语句导入的是模块对象的引用，而不是模块对象本身。因此，可以通过修改模块对象的属性或方法来实现重写方法的效果。例如：

import math

# 重写math.sqrt方法
def my_sqrt(x):
    return x ** 0.5

math.sqrt = my_sqrt

# 测试
print(math.sqrt(4))  # 输出2.0

在上面的代码中，我们首先使用import语句导入math模块，然后定义了一个函数my_sqrt，用于重写math.sqrt方法。接着，我们通过修改math.sqrt属性，将其指向新定义的my_sqrt函数。最后，我们测试了一下重写后的math.sqrt方法，输出结果为2.0。

需要注意的是，这种做法可能会引起意料之外的问题，特别是在多线程或多进程环境中。因为模块对象在解释器中是共享的，多个线程或进程可能会修改同一个模块对象，从而导致不可预测的结果。因此，不推荐在生产环境中使用这种技巧，而是应该通过继承或包装等方式实现方法的重写或扩展。