Python - ast - 如何找到第一个函数

Question

我试图在任意 Python 代码中找到第一个函数。

执行此操作的最佳方法是什么？

这是我到目前为止尝试过的方法。

import ast
import compiler.ast

code = """\
def func1():
    return 1

def func2():
    return 2

"""

tree = compiler.parse(code)

print list(ast.walk(tree))

但是我遇到了一个我不明白的错误。

Traceback (most recent call last):
  File "test.py", line 15, in <module>
    print list(ast.walk(tree))
  File "/usr/lib64/python2.7/ast.py", line 215, in walk
    todo.extend(iter_child_nodes(node))
  File "/usr/lib64/python2.7/ast.py", line 180, in iter_child_nodes
    for name, field in iter_fields(node):
  File "/usr/lib64/python2.7/ast.py", line 168, in iter_fields
    for field in node._fields:
AttributeError: Module instance has no attribute '_fields'

Answer 1

使用ast.parse , 而不是 compiler.parse:

>>> import ast
>>> 
>>> code = """
... def func1():
...     return 1
... 
... def func2():
...     return 2
... """
>>> 
>>> tree = ast.parse(code)
>>> [x.name for x in ast.walk(tree) if isinstance(x, ast.FunctionDef)]
['func1', 'func2']