确定在哪里执行功能?
问题内容:
我应该如何定义一个函数,where
该函数可以告诉它在哪里执行而没有传入任何参数?〜/ app /中的所有文件
a.py:
def where():
return 'the file name where the function was executed'
b.py:
from a import where
if __name__ == '__main__':
print where() # I want where() to return '~/app/b.py' like __file__ in b.py
c.py:
from a import where
if __name__ == '__main__':
print where() # I want where() to return '~/app/c.py' like __file__ in c.py
问题答案:
您需要使用来查找调用堆栈inspect.stack()
:
from inspect import stack
def where():
caller_frame = stack()[1]
return caller_frame[0].f_globals.get('__file__', None)
甚至:
def where():
caller_frame = stack()[1]
return caller_frame[1]