tree_sitter_python_parser_quickstart_with_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# Parse a string of code
11tree = parser.parse(bytes("""
12def foo():
13    if bar:
14        baz()
15""", "utf8"))
16
17# Print the root node of the syntax tree
18root_node = tree.root_node
19print(root_node.sexp())