为什么 python ast 不能检测到 function 中的 if 语句？

Question

我是 ASTs 的新手，因为我有统计学背景。 根据我的观察，Python 无法检测到以下代码的 if 语句 -

import ast
from pprint import pprint

tree = ast.parse("""
def add(a, b):
  return a + b

def subr(a,b):
    if 2>3:
        print("true")
    else:
        print("false")
    return 0
""")




for node in ast.iter_child_nodes(tree):
    print(isinstance(node, ast.If))

但是，如果没有function，则可以检测到if语句——

import ast
from pprint import pprint

tree = ast.parse("""
if 2>3:
    print("true")
else:
    print("false")
""")




for node in ast.iter_child_nodes(tree):
    print(isinstance(node, ast.If))

有人可以告诉我我以前的代码块有什么问题吗？

Answer 1

@Mathias R。 杰森是正确的。 iter_child_nodes()只能遍历直接子节点。 要遍历整个代码，应该使用ast.walk()