如何在antlr的python访问者中进行迭代

问题描述

我正在尝试使用 python 目标的访问者创建对象。对于以下语法规则：

prog : stat+ ;

我有这个用于创建相应对象的访问者代码：

def visitProg(self,ctx:ExprParser.ProgContext):
    stat_list = []
    for s in ctx.stat():
        stat_list.append(self.visit(s))
    node = ObjProg(stat_list)
    return node

上面的代码工作正常。但是，当我将规则更改为：

prog : s1=stat (',' s2=stat)* ;

对应的访问者代码应该是什么？我试着这样写，但我收到编译错误：

def visitProg(self,ctx:ExprParser.ProgContext):
    stat_list = []
    stat_list.append(self.visit(ctx.s1))
    for s in ctx.s2:
        stat_list.append(self.visit(s))
    node = ObjProg(stat_list)
    return node

解决方法

在规则中：

prog : s1=stat (',' s2=stat)* ;

s1 和 s2 都是单个 stat 上下文。所以代码 for s in ctx.s2 不正确。

如果您使用 += 运算符：

prog : stats+=stat (',' stats+=stat)* ;

那么你可以这样做：

def visitProg(self,ctx:ExprParser.ProgContext):
    stat_list = []
    for s in ctx.stats:
        stat_list.append(self.visit(s))
    return ObjProg(stat_list)

如果需要将第一个 stat 与其他的分开，您可以这样做：

prog : first_stat=stat (',' stats+=stat)* ;

然后是这样的：

def visitProg(self,ctx:ExprParser.ProgContext):
    stat_list = [self.visit(ctx.first_stat)]
    for s in ctx.stats:
        stat_list.append(self.visit(s))
    return ObjProg(stat_list)

也许您需要检查 ctx.stats is None，我不确定。