python如何获取javalang.ast.Node中的所有节点

如果你想获取 `javalang.ast.Node` 中的所有节点，可以使用递归的方式遍历整个抽象语法树（AST），并将每个节点添加到一个列表中。

以下是一个示例代码，演示如何使用递归的方式获取所有节点：

import javalang

def get_all_nodes(node):
    nodes = [node]
    for _, child_node in node:
        if isinstance(child_node, javalang.ast.Node):
            nodes += get_all_nodes(child_node)
    return nodes

# 示例代码
with open("example.java", "r") as f:
    source_code = f.read()

tree = javalang.parse.parse(source_code)
all_nodes = get_all_nodes(tree)

在上面的示例代码中，我们首先使用 `javalang.parse.parse` 方法解析源代码，生成一个抽象语法树。然后，我们调用 `get_all_nodes` 函数，将根节点传入该函数。该函数递归遍历所有子节点，并将其添加到 `nodes` 列表中。最后，该函数返回所有节点的列表。

需要注意的是，在实际使用过程中，你可能还需要根据具体的需求对该函数进行一些修改。例如，你可能需要过滤掉某些节点，或者只获取特定类型的节点等等。

相关问题

python如何如何打印javalang.ast.Node中所有节点名称

要打印`javalang.ast.Node`中所有节点名称，可以使用Python中的递归函数来遍历AST树，并且使用`type()`函数来获取每个节点的类型，然后打印出节点名称。

下面是一个示例代码：

import javalang

def print_node_names(node):
    print(type(node).__name__)
    for _, child_node in node.filter(javalang.tree.Node):
        print_node_names(child_node)

# 示例代码
source_code = '''
public class HelloWorld {
    public static void main(String[] args) {
        System.out.println("Hello, World!");
    }
}
'''

tree = javalang.parse.parse(source_code)
print_node_names(tree)

这里定义了一个递归函数`print_node_names()`，它接受一个`javalang.ast.Node`对象作为参数，并打印该节点的类型名称。然后，它遍历该节点的所有子节点，并对每个子节点调用`print_node_names()`函数。

在示例代码中，我们首先使用javalang库的`parse()`函数解析Java源代码并获取AST树。然后，我们调用`print_node_names()`函数来打印AST树中的所有节点类型名称。

python ast.filter过滤javalang.tree.MethodDeclaration及javalang.tree.ClassDeclaration

可以使用 Python 的 ast 模块来过滤 javalang.tree.MethodDeclaration 和 javalang.tree.ClassDeclaration。

具体实现可以参考下面的代码：

import javalang
import ast

class JavaNodeVisitor(ast.NodeVisitor):
    def __init__(self):
        self.class_nodes = []
        self.method_nodes = []

    def visit_ClassDeclaration(self, node):
        self.class_nodes.append(node)

    def visit_MethodDeclaration(self, node):
        self.method_nodes.append(node)

def filter_java_nodes(java_code):
    tree = javalang.parse.parse(java_code)
    visitor = JavaNodeVisitor()
    visitor.visit(tree)
    return visitor.class_nodes, visitor.method_nodes

使用示例：

java_code = """
public class HelloWorld {
    public static void main(String[] args) {
        System.out.println("Hello, World");
    }
}
"""

class_nodes, method_nodes = filter_java_nodes(java_code)
print(class_nodes)
print(method_nodes)

输出结果：

[ClassDeclaration(modifiers=[], name='HelloWorld', extends=None, implements=[], body=[MethodDeclaration(modifiers=[public, static], type=VoidType(), name='main', parameters=[FormalParameter(modifiers=[], type=ArrayType(qualified_name='java.lang.String'), variable_declarator_id=VariableDeclaratorId(), position=<Position line=2, column=33>)], throws=[], body=Block(statements=[ExpressionStatement(expression=MethodInvocation(select=FieldAccess(target=Name(qualifier=None, identifier='System', position=<Position line=3, column=9>), name='out', position=<Position line=3, column=15>), member='println', arguments=[StringLiteral(value='Hello, World', position=<Position line=3, column=21>)], position=<Position line=3, column=8>), position=<Position line=3, column=8>)], position=<Position line=2, column=38>), position=<Position line=2, column=9>)], position=<Position line=1, column=1>)], position=<Position line=1, column=1>)]
[MethodDeclaration(modifiers=[public, static], type=VoidType(), name='main', parameters=[FormalParameter(modifiers=[], type=ArrayType(qualified_name='java.lang.String'), variable_declarator_id=VariableDeclaratorId(), position=<Position line=2, column=33>)], throws=[], body=Block(statements=[ExpressionStatement(expression=MethodInvocation(select=FieldAccess(target=Name(qualifier=None, identifier='System', position=<Position line=3, column=9>), name='out', position=<Position line=3, column=15>), member='println', arguments=[StringLiteral(value='Hello, World', position=<Position line=3, column=21>)], position=<Position line=3, column=8>), position=<Position line=3, column=8>)], position=<Position line=2, column=38>), position=<Position line=2, column=9>)]

可以看到，我们成功地过滤出了 Java 代码中的类和方法节点，并返回了它们的列表。