Question

我正在尝试使用ast来打开.py文件，并为文件中的每个类提供我想要的属性。

但是，我无法按预期行事。

我希望能够做到

import ast

tree = ast.parse(f)
for class in tree:
    for attr in class:
        print class+" "+attr.key+"="+attr.value

例如;有点像ElementTree with XML。或者也许我背后有完全错误的想法，在这种情况下，是否有可能以另一种方式做到这一点（如果没有，我会写一些东西来做）。

Answer 1

比这复杂一点。您必须了解AST的结构和涉及的AST节点类型。另外，使用NodeVisitor类。尝试：

import ast

class MyVisitor(ast.NodeVisitor):
    def visit_ClassDef(self, node):
        body = node.body
        for statement in node.body:
            if isinstance(statement, ast.Assign):
                if len(statement.targets) == 1 and isinstance(statement.targets[0], ast.Name):
                    print 'class: %s, %s=%s' % (str(node.name), str(statement.targets[0].id), str(statement.value))

tree = ast.parse(open('path/to/your/file.py').read(), '')
MyVisitor().visit(tree)

有关详细信息，请参阅the docs。

使用Python的ast从类中获取属性

1 个答案: