python - 捕获文件名

Question

我正在编写一个 Python 3.4 脚本。在我的目录中:

main.py
core.py

在main.py中我有:

import core
print(core.status())

在core.py中我有:

def status():
    return filename

我想要导入core.status()的文件名。这里是 filename='main.py' 因为我在很多文件中使用 core.status() ，所以使用 __main__ 不好.

是否可以捕获导入另一个函数的文件名并将其打印在函数内，如我上面所解释的？

Answer 1

您可以从堆栈跟踪中提取此内容:

import traceback


def status():
    print(traceback.extract_stack(limit=2)[-2][0])

-2 只是表示堆栈的倒数第二个条目，即所谓的 status()。

我猜你所说的导入是指调用。