tree_sitter_python_parser_quickstart_syntax_tree.py

1import tree_sitter_python as tspython
2from tree_sitter import Language, Parser
3
4# Load the Python language grammar
5PY_LANGUAGE = Language(tspython.language())
6
7# Initialize the parser
8parser = Parser(PY_LANGUAGE)
9
10# Source code to parse
11src = b"def foo():\n  if True:\n    print('hello')"
12
13# Parse the source code
14tree = parser.parse(src)
15
16# Print the root node of the tree
17print(tree.root_node)
18
19# Access specific parts of the tree
20root_node = tree.root_node
21print(f"Root type: {root_node.type}")
22print(f"Children count: {root_node.child_count}")