python - 使用 ast 获取模块名称

Question

如何使用 ast 获取 python 模块名称？我尝试了以下方法来获取模块节点，但看起来它没有名称信息:

class v(ast.NodeVisitor):
    def visit_Module(self, node):
        print "Module : %s" % node

v().visit_Module(ast.parse(f.read(), filename))

我基本上需要实际的模块名称(如果它在包内，那么完整的名称，如 a.b.module)。

Answer 1

AFAIU，ast 中不存在此信息。更正式地说，Python 中 Module 的抽象语法是:

mod = Module(stmt* body)
    | Interactive(stmt* body)
    | Expression(expr body)

    -- not really an actual node but useful in Jython's typesystem.
    | Suite(stmt* body)

正如您所见，Module 只有一个主体，它是一个语句节点列表。没有名字。

请注意，您将 f.read() 传递给访问者。这会返回文件的内容，而无需真正知道或关心该文件的命名方式 - 您可以从哪里获取模块名称？

通过执行 Python 代码，您可以在 Python 代码内部使用 __name__ 和 __package__。