如何动态检测函数是否从循环内部被调用？

可以借助Python的inspect和dis模块实现这个需求——通过分析调用栈的帧信息和字节码，判断函数调用是否发生在循环结构里。

实现代码 #

import inspect
import dis

def called_from_loop():
    # 拿到调用当前函数的帧
    caller_frame = inspect.currentframe().f_back
    if not caller_frame:
        return False
    
    # 解析调用者的字节码指令
    bytecode = dis.Bytecode(caller_frame.f_code)
    current_instr_offset = caller_frame.f_lasti
    
    # 检查是否存在循环相关的字节码
    for instr in bytecode:
        # FOR_ITER对应for循环，WHILE_LOOP对应while循环
        if instr.opname in ('FOR_ITER', 'WHILE_LOOP'):
            # 当前指令偏移量大于等于循环起始偏移量，说明在循环内
            if instr.offset <= current_instr_offset:
                return True
    return False

def some_function():
    if called_from_loop():
        print("嘿，我是从循环里被调用的！")

# 测试代码
for i in range(0, 1):
    some_function()

some_function()

运行结果 #

嘿，我是从循环里被调用的！

说明 #

• 核心逻辑是通过inspect获取调用栈帧，再用dis拆解调用者的字节码，识别循环相关的指令标记。
• 这种方式依赖Python的字节码规范，不同Python版本的字节码指令可能有细微差异，使用时需要注意兼容性。
• 嵌套循环、生成器表达式里的调用也会被判定为“循环内调用”，符合常规场景的预期。

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