python ast.filter用法

在 Python 的 ast 模块中，ast.filter 函数用于过滤指定类型的节点。它接受两个参数：ast 树和一个函数，函数用于判断节点是否符合过滤条件。该函数返回一个生成器，可以用于遍历所有符合条件的节点。

下面是一个示例代码，演示了如何使用 ast.filter 过滤出所有的函数定义节点：

import ast

# 定义一个函数，用于判断节点是否为函数定义节点
def is_function(node):
    return isinstance(node, ast.FunctionDef)

# 解析 Python 代码，生成 ast 树
code = '''
def add(x, y):
    return x + y

def sub(x, y):
    return x - y
'''
tree = ast.parse(code)

# 使用 ast.filter 过滤函数定义节点
functions = ast.filter(tree, is_function)

# 遍历所有的函数定义节点
for func in functions:
    print('Function name:', func.name)
    print('Function args:', [arg.arg for arg in func.args.args])

在上面的示例代码中，我们首先定义了一个名为 `is_function` 的函数，用于判断节点是否为函数定义节点。然后解析了一段 Python 代码，生成了对应的 ast 树。接着使用 ast.filter 过滤出所有的函数定义节点，并遍历所有的函数定义节点，打印出函数的名称和参数列表。

需要注意的是，ast.filter 返回的是一个生成器，如果需要多次遍历符合条件的节点，需要将其转换为列表。例如：`functions = list(ast.filter(tree, is_function))`