tree_sitter_rust_grammar_parsing_and_syntax_tree_traversal.py

1import tree_sitter_rust as tsrust
2from tree_sitter import Language, Parser
3
4# Load the Rust language
5RUST_LANGUAGE = Language(tsrust.language())
6
7# Initialize the parser
8parser = Parser(RUST_LANGUAGE)
9
10# Parse a string of Rust code
11source_code = b"""
12fn main() {
13    println!("Hello, world!");
14}
15"""
16tree = parser.parse(source_code)
17
18# Get the root node and its children
19root_node = tree.root_node
20print(f"Root node type: {root_node.type}")
21
22# Access specific nodes (e.g., the function definition)
23function_node = root_node.children[0]
24print(f"Function node type: {function_node.type}")
25print(f"Function name: {source_code[function_node.child_by_field_name('name').start_byte:function_node.child_by_field_name('name').end_byte].decode('utf8')}")