tree-sitter-sql

import tree_sitter_sql from tree_sitter import Language, Parser # Initialize the SQL language from the installed grammar package SQL_LANGUAGE = Language(tree_sitter_sql.language()) # Create a parser and set its language parser = Parser() parser.set_language(SQL_LANGUAGE) # SQL code to parse (must be bytes) sql_code = b""" SELECT id, name FROM users WHERE age > 30 ORDER BY name ASC; """ # Parse the SQL code tree = parser.parse(sql_code) # Get the root node of the syntax tree root_node = tree.root_node # Print a basic representation of the tree (for demonstration) # In a real application, you would traverse the tree or use queries. print(f"Root Node Type: {root_node.type}") print(f"Root Node Text: {root_node.text.decode('utf8')}") # Example of finding a specific node type (e.g., 'select_statement') # This is a basic traversal; for complex patterns, Tree-sitter queries are used. select_statement_node = None for child in root_node.children: if child.type == 'select_statement': select_statement_node = child break if select_statement_node: print(f"Found 'select_statement' node. Text: {select_statement_node.text.decode('utf8')}") else: print("No 'select_statement' node found.")