如何优雅实现可选模块的类型检查与函数分发？

问题背景 #

需要根据对象的具体类型（来自多个可选导入模块）分发到对应处理函数，无法依赖鸭子类型，现有预加载模块的实现较为繁琐，需更优雅的写法或按需导入的方案。

方案一：简化预加载实现 #

通过调整配置结构、精准异常处理来优化原有预加载方案，提升可读性与简洁性：

import importlib
from operator import attrgetter

# 定义类型-处理函数映射配置
type_handlers = [
    (("module1", "class1"), fcn1),
    (("module2", "class2a"), fcn2),
    (("module2.sub1", "class2b"), fcn2),
    (("module3", "class3"), fcn3),
]

# 构建分发字典，仅加载存在的模块与类
dispatch_map = {}
for (mod_name, cls_name), handler in type_handlers:
    try:
        mod = importlib.import_module(mod_name)
        cls = attrgetter(cls_name)(mod)
        dispatch_map[cls] = handler
    except (ImportError, AttributeError):
        # 仅捕获模块/类不存在的异常，避免隐藏未知错误
        pass

def run_function(obj):
    for cls, handler in dispatch_map.items():
        if isinstance(obj, cls):
            return handler(obj)
    raise KeyError("No handler found for this object type")

优化点 #

• 拆分同处理函数的不同类配置，结构更清晰
• 替换裸except为具体异常类型，避免吞掉其他错误
• 简化分发字典构建逻辑，可读性更强

方案二：按需导入模块（延迟加载） #

如果可选模块较多且多数不会被用到，可采用按需导入方式，仅在检查对象类型时加载对应模块，避免预加载冗余资源：

import importlib

# 配置：类的完整限定名 -> 处理函数
type_handler_config = {
    "module1.class1": fcn1,
    "module2.class2a": fcn2,
    "module2.sub1.class2b": fcn2,
    "module3.class3": fcn3,
}

def run_function(obj):
    # 获取对象的完整类路径
    obj_type_full = f"{obj.__class__.__module__}.{obj.__class__.__name__}"
    
    # 优先匹配完整类路径
    if obj_type_full in type_handler_config:
        return type_handler_config[obj_type_full](obj)
    
    # 处理继承情况：遍历配置，按需加载类并检查isinstance
    for cls_path, handler in type_handler_config.items():
        mod_name, cls_name = cls_path.rsplit(".", 1)
        try:
            mod = importlib.import_module(mod_name)
            cls = getattr(mod, cls_name)
            if isinstance(obj, cls):
                return handler(obj)
        except (ImportError, AttributeError):
            continue
    
    raise KeyError("No handler found for this object type")

优势 #

• 无需预加载所有模块，节省初始化资源
• 支持直接匹配类的完整路径与继承关系检查
• 配置直观，直接使用类的完整限定名

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