如何将 XML 转换为 python AST答案

【问题标题】：How To Convert XML to python AST如何将 XML 转换为 python AST
【发布时间】：2018-12-28 02:05:53
【问题描述】：

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

    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 XMLtree_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 树，以便查看对实际源代码的影响。所以我的问题是：给定像'tree_a.xml'这样的xml文件，我如何将它转换为python AST树。我的目标是最终回到原始源代码。但是，如果我有来自 xml 文件的 python AST 树，我可以这样做。任何帮助表示赞赏。

【问题讨论】：

alpha5401 你有想过这个吗？我正在尝试做类似的事情 - 尝试编译和执行您的 'd=ast.dump(tree_a)' 生成的字符串。

标签： python xml syntax tree abstract-syntax-tree

【解决方案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 中可用）检查由 to_ast 创建的新 ast 实例是否与原始实例匹配：

print(ast.unparse(new_ast))

输出：

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

【讨论】：