如何在运行时将指定Python模块设置为不可导入/隐藏？

我在不同环境中安装了不同的依赖集（核心依赖全环境安装，仅用于代码检查的可选依赖、仅生产环境的可选依赖等）。本地环境通常会安装所有依赖，因此代码不会意外出现导入失败的情况，但生产环境中并未安装例如Mypy这类可选依赖。若我因临时需求（如绕过某个Bug）从这类可选依赖中导入内容，可能会忘记设置条件导入。

简而言之，我希望编写测试用例，强制将特定模块设置为不可用/不可导入/隐藏状态，无论测试运行环境如何，导入这些模块都会触发ImportError。是否有实现该需求的方法？比如能否注册某种钩子，在任意位置导入指定模块时主动抛出错误？

我已尝试修改sys.modules，但即使从中删除模块也无法阻止其被导入，删除模块本身也仅能使其暂时不可用，再次导入后又会恢复。

可以通过修改Python的导入机制钩子来实现，以下是两种实用方法：

方法一：自定义导入钩子（利用sys.meta_path） #

Python的sys.meta_path存储了导入查找的优先级钩子，我们可以添加一个自定义查找器，当检测到目标模块被导入时直接抛出ImportError。这种方法全局生效，优先级高于默认导入逻辑。

示例代码：

import sys

class BlockedModuleFinder:
    def __init__(self, blocked_modules):
        self.blocked_modules = blocked_modules

    def find_spec(self, fullname, path, target=None):
        if fullname in self.blocked_modules:
            raise ImportError(f"Module {fullname} is intentionally blocked")
        return None  # 交由其他查找器处理后续逻辑

# 配置要阻止的模块列表
blocked_modules = ['mypy', 'pylint']
sys.meta_path.insert(0, BlockedModuleFinder(blocked_modules))

# 测试验证
try:
    import mypy
except ImportError as e:
    print(e)  # 输出: Module mypy is intentionally blocked

方法二：给sys.modules设置“拦截模块” #

之前删除sys.modules条目无效，是因为Python会重新查找并导入模块。但如果我们给目标模块对应的sys.modules位置设置一个会抛出错误的对象，就能彻底阻止后续导入：

示例代码：

import sys

class BlockedModule:
    def __getattr__(self, name):
        raise ImportError("This module is blocked")

# 先清除已导入的模块（如果有的话）
if 'mypy' in sys.modules:
    del sys.modules['mypy']
# 设置拦截模块
sys.modules['mypy'] = BlockedModule()

# 测试验证
try:
    import mypy
    mypy.some_function()  # 访问模块属性也会触发ImportError
except ImportError as e:
    print(e)

注意：如果要阻止子模块（如mypy.api），需要单独给子模块设置拦截对象，或者在BlockedModule的__getattr__方法中统一拦截。

测试框架中的最佳实践 #

在pytest等测试框架中，可以用fixture临时添加/移除钩子，避免影响其他测试用例：

import sys
import pytest

@pytest.fixture
def block_mypy():
    # 保存原有导入钩子
    original_meta = sys.meta_path.copy()
    # 添加自定义拦截器
    class MypyBlocker:
        def find_spec(self, fullname, path, target=None):
            if fullname == 'mypy':
                raise ImportError("mypy is blocked for this test")
            return None
    sys.meta_path.insert(0, MypyBlocker())
    
    yield  # 执行测试用例
    
    # 恢复原有导入逻辑
    sys.meta_path = original_meta

使用该fixture的测试用例：

def test_condition_import(block_mypy):
    # 验证代码在mypy不可用时的逻辑
    with pytest.raises(ImportError):
        import mypy

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