tree_sitter_sql_parser_quickstart_with_syntax_tree_output.py

1import tree_sitter_sql as tssql
2from tree_sitter import Language, Parser
3
4# Initialize the SQL language from the installed tree-sitter-sql package
5SQL_LANGUAGE = Language(tssql.language())
6
7# Create a parser and set its language
8parser = Parser(SQL_LANGUAGE)
9
10# The SQL code to parse
11sql_code = "SELECT name, age FROM users WHERE age > 21;"
12
13# Parse the code into a syntax tree
14tree = parser.parse(bytes(sql_code, "utf8"))
15
16# Print the root node of the syntax tree
17print(tree.root_node)
18
19# Example: Print the s-expression of the tree
20print(tree.root_node.sexp())