使用tree_sitter获取代码AST抽象语法树_tree-sitter使用

1. 安装tree_sitter包

pip install tree_sitter

2. 下载语言支撑包

在当前工程目录下新建vendor目录用于存放语言支撑包，tree_sitter支持大部分主流语言，可以参考官网：Tree-sitter｜Introduction

如何需要支持java、python、C、C++、C#和javascript，需要在vendor目录下下载以下依赖包：

git clone https://github.com/tree-sitter/tree-sitter-java
git clone https://github.com/tree-sitter/tree-sitter-python
git clone https://github.com/tree-sitter/tree-sitter-c
git clone https://github.com/tree-sitter/tree-sitter-cpp
git clone https://github.com/tree-sitter/tree-sitter-c-sharp
git clone https://github.com/tree-sitter/tree-sitter-javascript

依赖包下载完毕后，需要根据依赖包生成语法解析器my-languages.so,此处创建build.py文件，文件内容如下：

#coding=utf-8
from tree_sitter import Language,Parser
import warnings
warnings.simplefilter('ignore', FutureWarning) 
 
Language.build_library(
  # Store the library in the `build` directory
  'build/my-languages.so',
  
  # Include one or more languages
  [
    'vendor/tree-sitter-c',
    'vendor/tree-sitter-cpp',
    'vendor/tree-sitter-python'
  ]
)

需要注意由于Language.build_library方法在tree_sitter的后期版本中将被替换，因此会出现以下告警信息：

tree_sitter/__init__.py:36: FutureWarning: Language(path, name) is deprecated. Use Language(ptr, name) instead.
  warn("{} is deprecated. Use {} instead.".format(old, new), FutureWarning)

 为了消除以上告警，在文件中添加以下语句：

import warnings
warnings.simplefilter('ignore', FutureWarning) 

在build目录中生成my-languages.so后，可以验证语法解析器的功能，定义main.py文件代码如下：

#coding=utf-8
from tree_sitter import Language, Parser
 
# 注意C++对应cpp，C#对应c_sharp（！这里短横线变成了下划线）
# 看仓库名称
CPP_LANGUAGE = Language('build/my-languages.so', 'cpp')
#CS_LANGUAGE = Language('build/my-languages.so', 'c_sharp')
 
# 举一个CPP例子
cpp_parser = Parser()
cpp_parser.set_language(CPP_LANGUAGE)
 
# 这是b站网友写的代码，解析看看
cpp_code_snippet = '''
int mian{
  piantf("hell world");
  remake O;
}
'''
 
# 没报错就是成功
tree = cpp_parser.parse(bytes(cpp_code_snippet, "utf8"))
# 注意，root_node 才是可遍历的树节点
root_node = tree.root_node
 
print(root_node)
print('ok')

输出语法树节点信息说明ok，语法树的详细解析可以参考官网的tree格式按需解析。