python - 获取包含调用函数的脚本路径

Question

假设我有两个文件，foo.py 和 bar.py。它们位于何处并不重要，只有 foo.py 以某种方式设法导入 bar.py 并调用定义的函数 func()在后者。我如何从 bar.func() 获取 foo.py 的绝对路径？

Answer 1

我不确定“从 bar.func() 获取 foo.py 的绝对路径”是什么意思。但是，函数可以通过使用 sys._getframe() 返回调用堆栈并检查调用函数的全局变量(变量 namespace 在下面的代码中)。这是一个显示如何执行此操作的示例:

foo.py

import bar

bar.func()

bar.py

from pprint import pprint
import sys

def func():
    namespace = sys._getframe(1).f_globals  # caller's globals
    pprint(namespace)
    print("caller's path:", namespace['__file__'])

示例输出:

{'__annotations__': {},
 '__builtins__': <module 'builtins' (built-in)>,
 '__cached__': None,
 '__doc__': None,
 '__file__': '\\Stack Overflow\\get-path-of-calling-script\\foo.py',
 '__loader__': <_frozen_importlib_external.SourceFileLoader object at 0x0085A7D0>,
 '__name__': '__main__',
 '__package__': None,
 '__spec__': None,
 'bar': <module 'bar' from '\\Stack Overflow\\get-path-of-calling-script\\bar.py'>,
 'sys': <module 'sys' (built-in)>}
caller's path: \Stack Overflow\get-path-of-calling-script\foo.py

看到返回字典中的'__file__'条目了吗？它是源文件的绝对路径 — foo.py — 包含对 bar.func() 的调用。