Python如何安全替换外部包全局使用的类且不改包源码

不需要修改Example包的源码，也不需要逐个重写包内依赖EgClass的函数、类，直接在运行时修改内存中加载的原始EgClass类对象即可，包内所有引用EgClass的位置都会自动使用修改后的版本。

实现逻辑 #

Python中导入的类本质是存在于模块命名空间的可修改对象，你不需要通过继承生成子类，直接给原始类绑定新增的方法、覆盖原有方法，所有基于这个类实例化的对象都会自动获得修改后的逻辑。

操作步骤 #

• 把补丁代码放在项目入口的最靠前位置，确保在导入Example包内其他模块（比如包含call1()的模块）之前执行
• 导入原始EgClass，把你要新增的方法直接绑定到类对象上
• 补丁打完后再正常导入、使用Example包的所有功能即可

代码示例 #

# 以下补丁代码必须优先执行，放在所有导入Example包其他模块的代码之前
from Example import EgClass

# 定义你要新增的方法，第一个参数固定为self
def new_method(self):
    # 填写你自己的方法逻辑
    pass

# 把方法直接绑定到原始类上
EgClass.new_method = new_method

# 之后正常导入包内的其他函数、类即可
from Example import call1

# 测试：call1内部实例化的EgClass对象，已经可以正常调用new_method
if __name__ == "__main__":
    call1()

注意点 #

• 如果需要重写EgClass的原有方法，操作逻辑完全一致：先把原方法存为临时变量（如果重写后还需要调用原逻辑的话），再用你写的新函数覆盖类上的同名属性即可
• 必须保证补丁执行顺序优先于其他依赖EgClass的模块导入，否则部分提前导入的模块会拿到未打补丁的原始类，导致补丁失效
• 这种修改是进程级别的，只在当前程序运行周期内生效，不会改动Example包本地磁盘上的源码，不会影响其他依赖这个包的项目

如果后续需要恢复类的原始状态，只需要把你绑定到类上的新增属性删除、把覆盖的原方法还原即可，常规开发场景下不需要做这步操作。

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