如何在python中使用tree sitter获得代码片段

通常来说,我们掌握解析字符串类型的代码内容就够了,如果有其他的数据类型的代码,也可以传递一个read的方法到parser的parse函数里(具体参考官方github的官方仓库吧)。注意,parser是基于bytes工作的,所以我们需要先把字符串类型转换为bytes,之后node内容的解析等都要记住在bytes上操作

code = f’{所有代码内容}’
src = bytes(code, ‘utf8’)
tree = parser.parse(src)

这样就很轻易地获得解析完的语法树了,是一个形如这样的结构,子节点我用缩进表示,兄弟节点就是同一层缩进的

program
comment
export statement

AST的使用场景

获得这样一棵语法树之后,通常会有以下几种使用场景

1. 看看树里的每个node是什么,以及node在代码原文中对应的片段是什么

这里要注意两个概念

  • node type:解析代码的时候实际上会对各种代码片段归类,比如说“statement block”这种node type,通常表示一个由花括号包裹的片段,不管是if,while还是函数体的花括号,里面可能都会包含一个子节点,它的node type都是"statement block"。
  • node text:每个node都会对应代码原文中的一段,父节点对应的片段会包含子节点对应的片段。注意,parser是在bytes上运作的,所以我们想要还原字符串的时候,还要从bytes还原回来:src_bytes[cursor.node.start_byte: cursor.node.end_byte].decode('utf-8')

举一个遍历语法树,打印node type 和 node text的例子

def display_tree(src_bytes, cursor, ident=4, node_text=True):
‘’’
display syntax node of the tree and its corresponding text

Args:
src_bytes: input bytes of the source code
cursor
ident: use to format the display info
node_text: whether to display the text of the node
‘’’
if node_text:
print(’ ’ * ident + cursor.node.type + ’ [’ + repr(
src_bytes[cursor.node.start_byte: cursor.node.end_byte].decode(‘utf-8’)) + ‘]’)
else:
print(’ ’ * ident + cursor.node.type)

if cursor.goto_first_child():
display_tree(src_bytes, cursor, ident + 4)
cursor.goto_parent()

while cursor.goto_next_sibling():
display_tree(src_bytes, cursor, ident)

cursor = tree.walk()
display_tree(src, cursor)

注意,遍历语法树使用到的cursor,是一个引用,在调用cursor.goto_parent()等方法时,会修改其指向的值,这一点在递归的时候要加以注意,在递归内部修改,也会影响到外部的cursor

此外,你可能会注意到,这个例子中使用了很多node的方法、cursor的方法,但是py-tree-sitter的github readme并没有给出很清晰的解说,你可以从这里看到更详细的,node / cursor 有哪些可供使用的方法、可供访问的属性
在这里插入图片描述

2. 使用query直接捕获树中的某一类节点

另一个需要掌握的使用场景是,不通过遍历的方式,直接捕获树上所有某种节点,比如我要拿到所有的函数调用、变量声明的代码片段。使用query时要注意,query是基于sexp来捕获的,它和我们上面遍历出来的各个node会有一点出入。sexp是形如 (program (comment) (export_statement declaration: (function_declaration name: (identifier)...)))的一种字符串表示,用括号来表示层级,附带一些形如declaration: 的子节点标识,这种标识在我们上面的遍历并打印node type的过程中一般是看不见的

因此,推荐想要使用query时,先使用print(tree.root_node.sexp()),看一下整体的情况,比如得到一个如下的内容在这里插入图片描述

再找到自己感兴趣的部分,这样使用query

自我介绍一下,小编13年上海交大毕业,曾经在小公司待过,也去过华为、OPPO等大厂,18年进入阿里一直到现在。

深知大多数Python工程师,想要提升技能,往往是自己摸索成长或者是报班学习,但对于培训机构动则几千的学费,着实压力不小。自己不成体系的自学效果低效又漫长,而且极易碰到天花板技术停滞不前!

因此收集整理了一份《2024年Python开发全套学习资料》,初衷也很简单,就是希望能够帮助到想自学提升又不知道该从何学起的朋友,同时减轻大家的负担。
img
img



既有适合小白学习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,基本涵盖了95%以上Python开发知识点,真正体系化!

由于文件比较大,这里只是将部分目录大纲截图出来,每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频,并且后续会持续更新

如果你觉得这些内容对你有帮助,可以添加V获取:vip1024c (备注Python)
img

做了那么多年开发,自学了很多门编程语言,我很明白学习资源对于学一门新语言的重要性,这些年也收藏了不少的Python干货,对我来说这些东西确实已经用不到了,但对于准备自学Python的人来说,或许它就是一个宝藏,可以给你省去很多的时间和精力。

别在网上瞎学了,我最近也做了一些资源的更新,只要你是我的粉丝,这期福利你都可拿走。

我先来介绍一下这些东西怎么用,文末抱走。


(1)Python所有方向的学习路线(新版)

这是我花了几天的时间去把Python所有方向的技术点做的整理,形成各个领域的知识点汇总,它的用处就在于,你可以按照上面的知识点去找对应的学习资源,保证自己学得较为全面。

最近我才对这些路线做了一下新的更新,知识体系更全面了。

在这里插入图片描述

(2)Python学习视频

包含了Python入门、爬虫、数据分析和web开发的学习视频,总共100多个,虽然没有那么全面,但是对于入门来说是没问题的,学完这些之后,你可以按照我上面的学习路线去网上找其他的知识资源进行进阶。

在这里插入图片描述

(3)100多个练手项目

我们在看视频学习的时候,不能光动眼动脑不动手,比较科学的学习方法是在理解之后运用它们,这时候练手项目就很适合了,只是里面的项目比较多,水平也是参差不齐,大家可以挑自己能做的项目去练练。

在这里插入图片描述

(4)200多本电子书

这些年我也收藏了很多电子书,大概200多本,有时候带实体书不方便的话,我就会去打开电子书看看,书籍可不一定比视频教程差,尤其是权威的技术书籍。

基本上主流的和经典的都有,这里我就不放图了,版权问题,个人看看是没有问题的。

(5)Python知识点汇总

知识点汇总有点像学习路线,但与学习路线不同的点就在于,知识点汇总更为细致,里面包含了对具体知识点的简单说明,而我们的学习路线则更为抽象和简单,只是为了方便大家只是某个领域你应该学习哪些技术栈。

在这里插入图片描述

(6)其他资料

还有其他的一些东西,比如说我自己出的Python入门图文类教程,没有电脑的时候用手机也可以学习知识,学会了理论之后再去敲代码实践验证,还有Python中文版的库资料、MySQL和HTML标签大全等等,这些都是可以送给粉丝们的东西。

在这里插入图片描述

这些都不是什么非常值钱的东西,但对于没有资源或者资源不是很好的学习者来说确实很不错,你要是用得到的话都可以直接抱走,关注过我的人都知道,这些都是可以拿到的。

一个人可以走的很快,但一群人才能走的更远。不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎扫码加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!
img

说确实很不错,你要是用得到的话都可以直接抱走,关注过我的人都知道,这些都是可以拿到的。**

一个人可以走的很快,但一群人才能走的更远。不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎扫码加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!
[外链图片转存中…(img-z4JbTgll-1712516171504)]

  • 27
    点赞
  • 29
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值