如何从调用模块获取文档字符串？

Question

我能得到主脚本的__doc__字符串吗？

这是起始脚本，它是来自命令行的运行：python a.py

模块a.py

import b
b.func()

模块b.py

def func():
    ???.__doc__

如何获取调用模块作为对象？

我不是在问如何获取字符串形式的文件名。我知道如何从堆栈跟踪中检索文件名。我不想通过手动解析来检索 doc 字符串。另外，由于循环导入循环，我认为我不能只按 m = __import__(a) 导入。

Answer 1

a.py

"""
Foo bar
"""

import b
if __name__ == '__main__':
    b.show_your_docs()

b.py

def show_your_docs():
    name = caller_name(1)
    print(__import__(name).__doc__)

其中 caller_name 是来自 gist

的代码

虽然这种方法的弱点是它获取模块名称的字符串表示并重新导入它，而不是获取对模块（类型）的引用。

Answer 2

这是@MatthewMartin 接受的答案的完整解决方案：

def show_your_docs():
    name = caller_name(1)
    print(__import__(name).__doc__)

def caller_docstring(level=1):
    name = caller_name(level)
    return __import__(name).__doc__

    def caller_name(skip=2):
        def stack_(frame):
            framelist = []
            while frame:
                framelist.append(frame)
                frame = frame.f_back
            return framelist

        stack = stack_(sys._getframe(1))
        start = 0 + skip
        if len(stack) < start + 1:
            return ''
        parentframe = stack[start]

        name = []
        module = inspect.getmodule(parentframe)
        if module:
            name.append(module.__name__)
        if 'self' in parentframe.f_locals:
            name.append(parentframe.f_locals['self'].__class__.__name__)
        codename = parentframe.f_code.co_name
        if codename != '<module>':  # top level usually
            name.append(codename)  # function or a method
        del parentframe
        return ".".join(name)

text = caller_docstring(2)

如何从调用模块获取文档字符串？

How do I get the docstring from a calling module?

python

docstring