Function get_node_by_kind

src/codetext/parser/language_parser.py:123–140

Get all nodes with specific type Args: root (tree_sitter.Node): Tree sitter root node kind (List[str]): (node's) type that want to get Return: List[tree_sitter.Node]: List of all

(root: tree_sitter.Node, kind: List[str])

def get_node_by_kind(root: tree_sitter.Node, kind: List[str]) -> List:
    """
    Get all nodes with specific type
    
    Args:
        root (tree_sitter.Node): Tree sitter root node
        kind (List[str]): (node's) type that want to get
    
    Return:
        List[tree_sitter.Node]: List of all 
    """
    assert type(root) == tree_sitter.Node, f"Expect `root` to be `tree_sitter.Node`, get {type(root)}"
    assert type(kind) in [list, str], f"Expect `kind` to be `list` of string or `str`, get {type(kind)}"
    assert all(isinstance(s, str) for s in kind) == True, f"Expect search kind to be `str`"

    node_list = []
    traverse_type(root, node_list, kind=kind)
    return node_list


def get_node_text(root: tree_sitter.Node) -> str: