从调用导入的 python 文件中获取名称

Question

我已经在互联网上搜索过，但到目前为止我找不到答案。

我有一个库，它被导入到所有其他文件中。假设库导入为 import my_lib 在使用它的文件中。

在 mylib 中我想做类似 __imported_from__ 的事情，这样代码就知道导入的来源。我需要这个用于记录目的（不同的调用不同 logfile/logname）

这将是目前的日志：

2018-09-13 01:36:00,921 - my_lib - INFO - Start Processing
2018-09-13 01:36:30,921 - my_lib - INFO - Done Processing
2018-09-13 01:37:00,921 - my_lib - INFO - Start Processing
2018-09-13 01:37:30,921 - my_lib - INFO - Done Processing

我希望 my_lib 包含导入文件中的名称，以便可以在我的日志记录中查看呼叫的来源。像下面的例子

2018-09-13 01:36:00,921 - import_from_file_1 - INFO - Start Processing
2018-09-13 01:36:30,921 - import_from_file_1 - INFO - Done Processing
2018-09-13 01:37:00,921 - import_from_file_2 - INFO - Start Processing
2018-09-13 01:37:30,921 - import_from_file_2 - INFO - Done Processing

根据答案，我写了 2 行，为我提供了我想要的数据：

import inspect
from os.path import basename, splitext
imported_from_file = (inspect.stack()[1][1]) if __name__ != '__main__' else False
used_from = splitext(basename(imported_from_file))[0] if imported_from_file else __name__

Answer 1

在正在导入的文件中，您可以检查堆栈并获取不以 < 开头的帧中的第一个文件名（在跳过第一帧后，即当前帧）表示一个内部库：

import inspect
if __name__ != '__main__':
    for frame in inspect.stack()[1:]:
        if frame.filename[0] != '<':
            print(frame.filename)
            break

如果您使用的是 Python 2.7，您应该改用框架元组的第二项。事实上，在 Python 2.7 中，执行导入的模块总是在第二帧中，所以你可以简单地做：

import inspect
if __name__ != '__main__':
    print(inspect.stack()[1][1])

从调用导入的 python 文件中获取名称

Get name from python file which called the import

python

python-import

python-2.7