Python如何强制始终使用本地模块而非库模块？含重导入及导入报错

问题背景 #

你在/src目录下存放了修改后的module.py，希望替代/some/other/path/library/module.py的库版本。直接用from module import TheClass能导入本地版本，但其他库代码重导入时仍会调用库模块；尝试用importlib修改sys.modules后，又出现ImportError: cannot import name 'TheClass' from 'module.TheClass'的错误。

问题根源 #

之前的importlib用法存在两个错误：

1. spec_from_file_location的第一个参数填了"module.TheClass"，这是错误的——该参数需要传入库模块的完整名称（即"library.subpack.module"）；
2. 没有执行模块加载逻辑，导致创建的模块对象是空的，自然无法找到TheClass。

正确解决方案 #

在myscript.py中执行以下代码，就能强制所有后续导入都使用本地修改的模块：

import sys
import importlib.util

# 定义目标库模块的完整名称
target_module_name = "library.subpack.module"
# 本地修改后的module.py路径
local_module_path = "./module.py"

# 创建模块规格对象
spec = importlib.util.spec_from_file_location(target_module_name, local_module_path)
# 创建空的模块对象
local_module = importlib.util.module_from_spec(spec)
# 执行本地模块的代码，加载类与内容
spec.loader.exec_module(local_module)
# 将本地模块注册到sys.modules，覆盖库版本的缓存
sys.modules[target_module_name] = local_module

# 测试导入，此时会使用本地版本
from library.subpack.module import TheClass
print(TheClass)

原理说明 #

• Python的sys.modules是全局模块缓存字典，所有导入操作都会优先查询这里的条目。将library.subpack.module对应的缓存值替换为你的本地模块后，后续任何地方的导入都会直接使用这个本地模块。
• spec.loader.exec_module(local_module)是关键步骤，它会执行module.py中的代码，把TheClass等内容加载到模块对象中，否则模块是空的，会触发导入错误。

备选方案（仅适用于短模块名导入场景） #

如果所有代码（包括库代码）都用import module而非完整路径导入，也可以通过修改sys.path优先级实现：

import sys
# 将src目录放到sys.path最前面，让Python优先查找本地模块
sys.path.insert(0, "./")

from module import TheClass

但该方案仅适用于短模块名导入的场景，对于库代码使用完整路径导入的情况，还是sys.modules替换的方法更可靠。

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