如何避免导入子模块时运行__init__.py或检测导入语句结构

好问题！这确实是Python包设计里常见的一个小痛点——默认情况下，不管你导入的是包的哪个子模块，包根目录的__init__.py都会被执行。不过我们有两种靠谱的解决思路：

方法1：通过检测导入上下文条件执行__init__.py代码 #

你可以在__init__.py里通过检查调用栈，判断用户是直接导入包本身，还是导入子模块，从而决定是否执行默认的导入语句。

修改你的__init__.py如下：

import sys

def _is_importing_submodule_directly():
    # 获取调用栈的上一级帧，判断导入行为的上下文
    try:
        caller_frame = sys._getframe(1)
    except ValueError:
        # 没有上一级帧（比如直接运行__init__.py），默认执行导入
        return False
    
    # 判断调用者是否是mypackage的子模块（不是mypackage本身）
    caller_module_name = caller_frame.f_globals.get('__name__', '')
    return caller_module_name.startswith('mypackage.') and caller_module_name != 'mypackage'

# 只有当用户直接导入mypackage时，才执行默认导入
if not _is_importing_submodule_directly():
    from mypackage.moduleA import abc
    from mypackage.moduleB import jkl

原理说明 #

• sys._getframe(1)获取调用当前__init__.py的上一级栈帧，通过帧的全局变量判断导入的模块名。
• 如果用户执行from mypackage.oldModule import xyz，调用者的模块名是mypackage.oldModule，这时我们就跳过默认的导入代码。
• 注意：sys._getframe()是CPython的私有API，虽然在绝大多数场景下都能正常工作，但如果你的包需要兼容其他Python解释器（比如PyPy），可能需要谨慎使用。

方法2：重构包结构，隔离旧模块（推荐） #

如果可以调整包的结构，把旧模块放到一个独立的子包里，是更干净且符合Python规范的解决方案。

调整后的包结构：

/mypackage
|-- __init__.py
|-- moduleA.py
|-- moduleB.py
|-- legacy/
    |-- __init__.py  # 可以留空，不需要导入任何内容
    |-- oldModule.py

此时：

• 普通用户依然可以用import mypackage as mpkg直接访问mpkg.abc()和mpkg.jkl()，根目录__init__.py保持原来的导入逻辑不变。
• 需要旧模块的用户需要执行from mypackage.legacy.oldModule import xyz，这时只会触发legacy/__init__.py的执行（而它是空的），不会运行根目录的__init__.py代码。

为什么推荐这种方法 #

• 完全避免了条件判断的复杂性，包结构更清晰，用户能直观区分核心功能和遗留功能。
• 不依赖任何私有API，兼容性更好，也更符合Python的包设计最佳实践。

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