如何不修改原文件替换Python导入类（第三方包场景）

可以通过Python的模块导入机制实现，核心是在导入目标类B之前，将自定义的A类注入到b.py依赖的xxx模块命名空间中，让b.py在导入A时获取到我们的自定义版本。

方案一：替换已有xxx模块中的A类 #

如果xxx模块是真实存在的，我们可以先导入该模块，再替换其中的A属性：

class A:
    def __init__(self):
        print("init my own A")

# 关键步骤：在导入B前修改xxx模块的A类
import sys
import xxx

# 将自定义A替换掉xxx模块中的原A
xxx.A = A

# 此时导入B，b.py里的from xxx import A会拿到自定义的A
from somepackage.b import B

b = B()

方案二：模拟xxx模块（适用于原模块不存在或不想加载原模块） #

如果不想加载原xxx模块，可以直接创建一个模拟的xxx模块并注入到Python的模块缓存中：

class A:
    def __init__(self):
        print("init my own A")

import sys
import types

# 创建一个名为xxx的模块对象
xxx_module = types.ModuleType('xxx')
# 将自定义A绑定到该模块
xxx_module.A = A
# 把模块放入sys.modules，后续导入xxx时会优先使用这个模块
sys.modules['xxx'] = xxx_module

# 导入B
from somepackage.b import B

b = B()

注意事项 #

• 必须在导入B之前完成替换操作：Python模块只会被导入一次，一旦b.py已经加载，再修改xxx.A不会对已加载的B类生效。
• 修复原代码的语法问题：你提供的b.py和自定义A的__init__方法都缺少self参数，实际运行时会触发TypeError，需要补充该参数。

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