Question

我希望有人可以帮助我在此处找到正确的资源。我在这个论坛和其他地方四处张望，但找不到合适的资源。抱歉，我的问题不清楚，所以让我在这个简单的例子中说明一下：原始源代码：

    a='''
for x in y:
    print(x)
'''

此代码的python AST树是：

tree_a=ast.parse(a)

这棵树的平面版本：

d=ast.dump(tree_a)

输出（d的内容）：

"Module(body=[For(target=Name(id='x', ctx=Store()), iter=Name(id='y', ctx=Load()), body=[Expr(value=Call(func=Name(id='print', ctx=Load()), args=[Name(id='x', ctx=Load())], keywords=[]))], orelse=[])])"

使用此资源Python AST to XML tree_a可以转换为XML格式：

res_0 = ast2xml().convert(tree_a)
with open('tree_a.xml','w') as f:
    res1=f.write(str(prettify(res_0)))

“ tree_a.xml”的内容为：

<?xml version="1.0" ?>
<ast>
  <Module>
    <body>
      <For col_offset="0" lineno="2">
        <Name col_offset="4" id="x" lineno="2">
          <Store/>
        </Name>
        <Name col_offset="9" id="y" lineno="2">
          <Load/>
        </Name>
        <body>
          <Expr col_offset="4" lineno="3">
            <Call col_offset="4" lineno="3">
              <Name col_offset="4" id="print" lineno="3">
                <Load/>
              </Name>
              <args>
                <Name col_offset="10" id="x" lineno="3">
                  <Load/>
                </Name>
              </args>
              <keywords/>
            </Call>
          </Expr>
        </body>
        <orelse/>
      </For>
    </body>
  </Module>
</ast>

以xml格式存储树，我想对文件进行一些测试和修改，然后返回到python AST树，以查看对实际源代码的影响。所以我的问题是：给定xml文件，例如“ tree_a.xml”，我如何将其转换为python AST树。我的目标是最终返回到原始源代码。但是，如果我从xml文件中获得了python AST树，则可以这样做。任何帮助表示赞赏。

Answer 1

每个 ast literal 都包含一个 _fields 属性，您可以使用该属性通过递归重建原始 ast 对象：

import ast, xml.etree.ElementTree as ET
from itertools import zip_longest as zl
def to_ast(xml):
   if xml is None:
      return {}
   if not hasattr(ast, xml.tag):
      return [to_ast(i) for i in xml]
   return (t:=getattr(ast, xml.tag))(**(k:=xml.attrib), 
          **{a:to_ast(b) for a, b in zl([i for i in t._fields if i not in k], xml)})

xml_tree = ET.parse('tree_a.xml')
new_ast = to_ast(list(xml_tree._root)[0])

您可以使用 ast.unparse（在 Python 3.9 中可用）来检查由 ast 创建的新 to_ast 实例是否与原始实例匹配：

print(ast.unparse(new_ast))

输出：

for x in y: # type: {}
    print(x)

如何将XML转换为python AST

1 个答案: