在python中检查堆栈中的局部变量

Question

我写了一个小函数，它在堆栈中查找一层并查看其中是否有变量。但是如何将这个函数变成一个在堆栈中一直查找直到找到一个局部变量的函数？

import inspect
def variable_lookup(variable):
    parent_locals = inspect.currentframe().f_back.f_locals
    if variable in parent_locals.keys():
        return parent_locals[variable]

我想要的是循环，直到它遍历所有堆栈，试图找到变量。我如何使用检查来做到这一点？有没有更简单的方法？

我从这个开始，但是 f_back 没有返回一个框架对象，所以我不知道从这里去哪里：

import inspect
def variable_lookup(variable):
previous_frame = inspect.currentframe().f_back
while True:
    print(variable in previous_frame.f_locals.keys())
    previous_frame = inspect.getframeinfo(previous_frame)

因此代码在AttributeError: 'Traceback' object has no attribute 'f_locals' 下失败

Answer 1

有一个更高级别的函数可用：inspect.stack()。这将为您提供FrameInfo() 对象的列表，这些对象具有frame 引用：

for frameinfo in inspect.stack(0):
    if variable in frameinfo.frame.f_locals:
        return frameinfo.frame.f_locals[variable]

我将源代码行数（context 参数）设置为零，加载源代码行只是为了检查局部变量。

stack()函数基本实现为：

return inspect.getouterframes(inspect.currentframe(), context)

和getouterframes() 只是跟随frame.f_back 引用，直到找到None。所以你也可以使用while 循环来实现它：

frame = inspect.currentframe()
while frame:
    if variable in frame.f_locals:
        return frame.f_locals[variable]
    frame = frame.f_back

这当然更轻，因为这避免了为堆栈中的所有帧创建FrameInfo() 实例，避免了对所有帧的文件名的自省，并且如果尽早找到变量，则不需要访问所有链接的帧。