在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').如何才能做到这一点？

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

编辑:这里有一些代码可以做你想要的,我想:

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)