Handle FuncCall better in visit_replace

This commit is contained in:
Simon Lindholm
2022-07-06 14:42:13 +02:00
parent fbfff44f4f
commit 05d27acb95
+3 -1
View File
@@ -286,7 +286,9 @@ def visit_replace(top_node: ca.Node, callback: Callable[[ca.Node, bool], Any]) -
node.left = rec(node.left)
node.right = rec(node.right)
elif isinstance(node, ca.FuncCall):
# not worth replacing .name
# not worth replacing .name if it's a normal function call
if not isinstance(node.name, ca.ID):
node.name = rec(node.name)
if node.args:
rec(node.args, True)
elif isinstance(node, ca.ExprList):