support for traversing AST tree with parens and objectcomps

This commit is contained in:
Jameson Lee 2018-05-30 23:39:41 +00:00 committed by Dave Cunningham
parent 46256581d7
commit a1964b49f1

View File

@ -263,7 +263,7 @@ func specialChildren(node ast.Node) []ast.Node {
case *ast.ArrayComp: case *ast.ArrayComp:
return []ast.Node{node.Body} return []ast.Node{node.Body}
case *ast.ObjectComp: case *ast.ObjectComp:
return inObjectFieldsChildren(node.Fields)
case *ast.Self: case *ast.Self:
return nil return nil
case *ast.SuperIndex: case *ast.SuperIndex:
@ -274,6 +274,8 @@ func specialChildren(node ast.Node) []ast.Node {
return nil return nil
case *ast.Var: case *ast.Var:
return nil return nil
case *ast.Parens:
return nil
} }
panic(fmt.Sprintf("specialChildren: Unknown node %#v", node)) panic(fmt.Sprintf("specialChildren: Unknown node %#v", node))
} }