def top_level_classes(body):
# return (node for node in ast.walk(body) if isinstance(node, ast.ClassDef))
return (f for f in body if isinstance(f, ast.ClassDef))
微信
公众号
扫码关注公众号