python - 我如何找出在哪个文件或模块中调用了我的函数？

Question

Python 2.5 到 2.7:

#a.py:
def foo():
    pass

#b.py
from a import foo
foo()

从foo()，我想知道它已经在“b”模块中被调用了。我现在能想到的唯一方法是引发异常，捕获它并检查回溯(向上一级)。有没有一种自然的方法可以做到这一点？

Answer 1

您可以使用检查模块执行此操作。

例如

#!/usr/bin/env python
# a.py
import inspect

def foo():
    for item in inspect.stack():
        print item

-

#!/usr/bin/env python
# b.py

from a import foo

foo()

-

$ python b.py
(<frame object at 0x2026fb0>, '/home/tdb/a.py', 6, 'foo', ['    for item in inspect.stack():\n'], 0)
(<frame object at 0x1fe4a30>, 'b.py', 5, '<module>', ['foo()\n'], 0)