如何通过反射从Python包加载实例化模型类？解决下标访问报错

这个问题我之前也碰到过，核心原因是你把模块对象当成了字典来用——models是Python的模块实例，不是可下标访问的容器，所以models[model]会抛出TypeError: 'module' object is not subscriptable。下面给你几个优雅的解决方案，按推荐程度排序：

方案1：在__init__.py中维护模型映射字典（最可控） #

这种方式最清晰，也能明确控制哪些模型可以被外部调用，避免误访问模块内的其他属性。

修改models/__init__.py：

from models.model_a import ModelA
from models.model_b import ModelB

# 显式定义模型名称到类的映射
MODEL_MAP = {
    "ModelA": ModelA,
    "ModelB": ModelB
}

然后在main.py中修改get_model函数：

import models
model = get_model(config.use_model) # config.use_model == "ModelA"

def get_model(model_name):
    # 可选：添加模型存在性检查，避免KeyError
    if model_name not in models.MODEL_MAP:
        raise ValueError(f"不支持的模型类型：{model_name}")
    return models.MODEL_MAP[model_name](**config[model_name].structure)

优点：直观可控，能清晰看到所有可调用的模型；可以轻松扩展新模型（只需在字典里加一行）；避免误访问模块内的其他变量。

方案2：使用getattr直接获取模块属性（最简洁） #

如果不想修改__init__.py，可以用Python内置的getattr函数，直接从模块中按名称获取模型类：

修改main.py的get_model函数：

import models
model = get_model(config.use_model) # config.use_model == "ModelA"

def get_model(model_name):
    try:
        model_cls = getattr(models, model_name)
        return model_cls(**config[model_name].structure)
    except AttributeError:
        raise ValueError(f"不支持的模型类型：{model_name}")

优点：无需修改__init__.py，代码简洁；
缺点：依赖模型类名和配置中的字符串完全一致，容易因拼写错误出问题；如果模块内有同名的非模型属性，可能会误获取。

方案3：自动收集模型类（适合大量模型的场景） #

如果你的模型数量很多，手动维护映射字典太麻烦，可以用inspect模块自动收集模块内的模型类：

修改models/__init__.py：

import inspect
from models.model_a import ModelA
from models.model_b import ModelB

# 自动收集当前模块下的所有类（过滤掉非模型类）
MODEL_MAP = {
    name: cls for name, cls in locals().items()
    if inspect.isclass(cls) and cls.__module__ == __name__
}

然后main.py的用法和方案1完全一致。

优点：新增模型时无需手动修改映射字典，自动生效；
缺点：需要确保所有模型类都在当前模块的命名空间中，且要注意过滤掉不需要的类（比如导入的其他工具类）。

额外建议 #

不管用哪种方案，都建议添加模型存在性检查，抛出明确的错误信息，这样调试起来更方便，也能避免程序因未处理的异常崩溃。

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