tree_sitter_python_parser_init_and_syntax_tree_traversal.py

1import tree_sitter_python as tspython
2from tree_sitter import Language, Parser
3
4# Get the language object for Python
5PY_LANGUAGE = Language(tspython.language())
6
7# Initialize the parser
8parser = Parser(PY_LANGUAGE)
9
10# Parse some source code
11tree = parser.parse(bytes("""
12def foo():
13    if bar:
14        baz()
15""", "utf8"))
16
17# Access the root node
18root_node = tree.root_node
19
20# Inspect the tree
21assert root_node.type == "module"
22assert root_node.start_point == (1, 0)
23assert root_node.end_point == (5, 0)
24
25# Traverse the children
26function_node = root_node.children[0]
27assert function_node.type == "function_definition"
28assert function_node.child_by_field_name("name").text == b"foo"