如何在Python中监听特定包导入并触发指定函数？

核心思路 #

Python允许通过修改sys.meta_path添加自定义导入钩子，精准捕获目标模块的导入事件。核心逻辑分两步：

1. 先检查sys.modules，如果目标包已被导入，直接执行指定函数
2. 如果未导入，注册一个导入钩子，当目标包被导入时自动触发函数，且确保只执行一次

完整实现代码 #

import sys
import importlib.abc

def do_things_with_some_package():
    # 这里写你的慢逻辑，比如初始化绑定、资源加载等
    print("检测到some_package导入，执行慢函数")

class SomePackageImportHook(importlib.abc.MetaPathFinder):
    def find_spec(self, fullname, path, target=None):
        if fullname == "some_package":
            # 触发回调函数
            do_things_with_some_package()
            # 执行后从meta_path移除钩子，避免重复触发
            sys.meta_path.remove(self)
        # 返回None，让Python默认导入机制处理模块加载
        return None

def setup_some_package_callback():
    # 先检查模块是否已存在
    if "some_package" in sys.modules:
        do_things_with_some_package()
    else:
        # 注册导入钩子
        hook = SomePackageImportHook()
        sys.meta_path.append(hook)

# 在你的库初始化逻辑中调用这个函数
setup_some_package_callback()

关键细节说明 #

• sys.meta_path中的finder会在Python导入任何模块时被依次调用，我们的find_spec方法仅关注目标模块名，不影响其他模块的正常导入
• 回调执行后立刻移除钩子，确保do_things_with_some_package只运行一次，避免重复执行慢逻辑
• 完全覆盖两种场景：
  • 你的库晚于some_package导入：直接检测到sys.modules中的模块，立即执行函数
  • 你的库早于some_package导入：钩子等待目标模块被导入时自动触发函数

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