如何高效实现多模块一站式索引导入并避免冗余加载

问题背景 #

现有Python项目结构如下：

├── program.py
└── interfaces
    ├── domain1.py
    ├── domain2.py
    └── index.py

interfaces/index.py原本直接导入所有domain模块，program.py通过导入index实现一站式命名空间调用，但即使未使用依赖torch（导入缓慢）的domain2，程序仍会加载torch导致启动缓慢。希望保留from interfaces import index as id; id.domain1.print_local()的调用方式，同时避免加载未使用的模块。

解决方案 #

方案1：利用__getattr__实现懒加载 #

修改interfaces/index.py，通过模块的__getattr__方法实现按需导入，只有当访问对应属性时才真正加载模块：

# interfaces/index.py
def __getattr__(name):
    if name == "domain1":
        from . import domain1
        return domain1
    elif name == "domain2":
        from . import domain2
        return domain2
    raise AttributeError(f"module 'interfaces.index' has no attribute '{name}'")

这种方式完全兼容原调用写法，未被访问的模块不会触发导入，自然也不会加载torch这类重依赖。

方案2：用类容器+属性懒加载 #

如果需要更清晰的结构，可以定义类作为命名空间容器，通过@property实现懒加载：

# interfaces/index.py
class Index:
    @property
    def domain1(self):
        from . import domain1
        return domain1
    
    @property
    def domain2(self):
        from . import domain2
        return domain2

index = Index()

此时program.py的调用方式调整为：

from interfaces.index import index as id
id.domain1.print_local()

同样能实现按需加载，且结构更直观。

方案3：通用动态导入方案 #

借助importlib模块实现通用化的动态导入，新增domain模块时无需修改index.py：

# interfaces/index.py
import importlib

def __getattr__(name):
    try:
        return importlib.import_module(f".{name}", __name__)
    except ImportError:
        raise AttributeError(f"module 'interfaces.index' has no attribute '{name}'")

该方案自动匹配同目录下的模块，扩展性更强。

注意事项 #

• 所有方案均保留了一站式命名空间的调用体验，无需直接导入单个domain模块
• 懒加载会在首次访问模块时产生微小延迟，但避免了启动时加载所有依赖的开销
• 若模块间存在循环依赖，需额外调整导入逻辑避免报错

内容的提问来源于stack exchange，提问作者Fabrice E.