Python中如何获取模块导入来源，根据导入方实现动态行为？

Python完全支持查询模块导入来源，也可以在模块内部识别导入方身份，实现你需要的动态继承逻辑，具体实现方法和注意点如下：

获取导入方模块名 #

模块第一次被导入时，顶层代码会被执行，此时调用栈中直接留存了触发import语句的帧信息，用标准库即可拿到导入方的模块名，不需要第三方依赖：

• 通用跨实现方案（兼容CPython、PyPy等主流Python实现）：

import inspect
caller_frame = inspect.currentframe().f_back
# 拿到导入方的模块全限定名
caller_module = caller_frame.f_globals.get('__name__', '')

• 轻量方案（仅CPython环境可用，性能更好）：

import sys
# 栈深度1就是直接触发当前模块导入的调用帧
caller_module = sys._getframe(1).f_globals.get('__name__', '')

注意：必须在模块顶层代码执行阶段抓取导入方信息，不要放到类方法、函数内部调用，否则调用栈已经发生变化，拿到的不是最初的导入方。

实现动态继承逻辑 #

拿到导入方模块名后，直接在顶层判断、动态选择父类定义类即可，完全匹配你提出的需求：

# 先导入两个需要切换的父类
from src.bases import MyParent, MyParent2

# 抓取导入方信息
import sys
caller_module = sys._getframe(1).f_globals.get('__name__', '')

# 配置需要触发父类切换的模块集合
SPECIAL_IMPORT_MODULES = {"auth.service", "order.trade_logic"}

# 动态选择父类
selected_base = MyParent2 if caller_module in SPECIAL_IMPORT_MODULES else MyParent

# 定义目标类
class MyClass(selected_base):
    # 原有类的所有属性、方法逻辑保持不变即可
    def your_method(self):
        pass

踩坑提示 #

• Python会对已导入的模块做缓存，存在sys.modules字典中，模块顶层代码只会在第一次被导入时执行一次。如果你的模块先被普通模块导入、后被特殊模块导入，第二次导入会直接取缓存的模块对象，不会重新执行顶层逻辑切换父类。如果需要支持不同导入方拿到不同的类实现，需要额外自定义模块级__getattr__或者导入钩子，常规场景下首次导入判断的逻辑已经足够。
• 这类动态逻辑会提升代码的理解和调试成本，仅建议在插件系统、多版本兼容层等必须动态适配的场景使用，常规业务逻辑不要滥用。

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