importlib.import_module() 的用法与实战案例

🌟 一、什么是 importlib？

importlib 是 Python 的一个内置标准库，用于在程序运行时 动态导入模块。

🔤 对比：普通 import vs importlib

✅ importlib 让你可以在 运行时 才决定导入哪个模块，而不是写死在代码里。

🧰 二、importlib 的核心函数：import_module

✅ 语法：

import importlib

module = importlib.import_module(module_name)

• module_name：一个字符串，表示模块名（如 "json"、"os.path"、"langchain.document_loaders"）
• 返回值：对应的模块对象，可以像普通 import 一样使用

🎯 三、为什么要用 importlib？—— 实际场景

假设你正在做一个 文档处理系统，支持多种文件类型：

你想根据文件后缀自动选择加载器，但你不想写一堆 if-elif：

if ext == "pdf":
    from langchain.document_loaders import PDFLoader
    loader = PDFLoader(path)
elif ext == "csv":
    from langchain.document_loaders import CSVLoader
    loader = CSVLoader(path)
...

这时候就可以用 importlib 实现 通用加载逻辑！

🧪 四、完整例子：动态加载文档加载器

import importlib
import os

def get_loader(file_path: str):
    """
    根据文件扩展名，动态选择并返回对应的文档加载器实例。
    """
    # 获取文件扩展名
    ext = os.path.splitext(file_path)[-1].lower()

    # 定义扩展名到加载器类名的映射
    loader_map = {
        ".pdf": "PyPDFLoader",           # 假设有这个类
        ".csv": "CSVLoader",
        ".json": "JSONLoader",
        ".txt": "TextLoader",
    }

    # 查找对应的 loader 类名
    loader_name = loader_map.get(ext)
    if not loader_name:
        raise ValueError(f"不支持的文件类型: {ext}")

    try:
        # 动态导入 langchain.document_loaders 模块
        module = importlib.import_module("langchain.document_loaders")
        
        # 从模块中获取类对象
        LoaderClass = getattr(module, loader_name)
        
        # 实例化并返回
        return LoaderClass(file_path)

    except ImportError:
        raise ImportError(f"无法导入 langchain.document_loaders，请安装 langchain")
    except AttributeError:
        raise ImportError(f"加载器 {loader_name} 不存在，请检查类名是否正确")

# === 使用示例 ===
if __name__ == "__main__":
    # 假设这些类在 langchain.document_loaders 中存在
    loader = get_loader("data/sample.pdf")
    print(f"使用的加载器: {loader.__class__.__name__}")

    loader = get_loader("data/users.csv")
    print(f"使用的加载器: {loader.__class__.__name__}")

✅ 输出可能为：

使用的加载器: PyPDFLoader
使用的加载器: CSVLoader

🔍 五、代码解析

💡 这样就实现了“通过字符串创建类对象”的效果，非常灵活！

🛠️ 六、其他常见用途

1. 插件系统（Plugin System）

# plugins/json_exporter.py
def export(data):
    print("导出为 JSON")

# 主程序动态加载插件
plugin_name = "json_exporter"
module = importlib.import_module(f"plugins.{plugin_name}")
module.export(data)

2. 配置驱动加载

config = {"loader": "CSVLoader", "file": "data.csv"}
loader_name = config["loader"]
Loader = getattr(importlib.import_module("langchain.document_loaders"), loader_name)
loader = Loader(config["file"])

3. 热重载开发（调试用）

import mymodule
importlib.reload(mymodule)  # 修改代码后重新加载，不用重启程序

⚠️ 七、注意事项

✅ 八、总结

📌 一句话记住它：

importlib 让你能用字符串来“导入模块”，把“代码逻辑”和“具体实现”解耦，是构建灵活系统的关键工具。