获取Python中调用函数模块的__name__

Question

假设 myapp/foo.py 包含：

def info(msg):
    caller_name = ????
    print '[%s] %s' % (caller_name, msg)

并且 myapp/bar.py 包含：

import foo
foo.info('Hello') # => [myapp.bar] Hello

我希望将 caller_name 设置为 __name__<在本例中，调用函数模块（即“myapp.foo”）的 /code> 属性。 如何才能做到这一点？

Answer 1

查看检查模块：

inspect.stack() 将返回堆栈信息。

在函数内部，inspect.stack()[1] 将返回调用者的堆栈。 从那里，您可以获得有关调用者的函数名称、模块等的更多信息。

有关详细信息，请参阅文档：

http://docs.python.org/library/inspect.html

另外，Doug Hellmann 在他的 PyMOTW 系列中对检查模块有一篇很好的文章：

http://pymotw.com/2/inspect/index.html#module-inspect

编辑：这里有一些代码可以完成您的任务我想：

import inspect 

def info(msg):
    frm = inspect.stack()[1]
    mod = inspect.getmodule(frm[0])
    print '[%s] %s' % (mod.__name__, msg)

Answer 2

面对类似的问题，我发现 sys 模块中的 sys._current_frames() 包含有趣的信息，可以为您提供帮助，而无需导入检查，至少在特定用例中是这样。

>>> sys._current_frames()
{4052: <frame object at 0x03200C98>}

然后，您可以使用 f_back “向上移动”：

>>> f = sys._current_frames().values()[0]
>>> # for python3: f = list(sys._current_frames().values())[0]

>>> print f.f_back.f_globals['__file__']
'/base/data/home/apps/apricot/1.6456165165151/caller.py'

>>> print f.f_back.f_globals['__name__']
'__main__'

对于文件名，您也可以使用 f.f_back.f_code.co_filename，如上面 Mark Roddy 所建议的。 我不确定此方法的限制和注意事项（多线程很可能是一个问题），但我打算在我的情况下使用它。

Answer 3

我不建议这样做，但您可以通过以下方法完成您的目标：

def caller_name():
    frame=inspect.currentframe()
    frame=frame.f_back.f_back
    code=frame.f_code
    return code.co_filename

然后按如下方式更新现有方法：

def info(msg):
    caller = caller_name()
    print '[%s] %s' % (caller, msg)

Answer 4

对于我来说，下面的行足以获取呼叫者的姓名。

import inspect
frame = inspect.stack()[-1]
print(frame.filename)