Python使用ElementTree美化XML格式-白红宇

Python使用ElementTree美化XML格式

阅读量：5112 次

发布时间：2019-06-13

本文共 2489 字，大约阅读时间需要 8 分钟。

Python中使用ElementTree可以很方便的处理XML，但是产生的XML文件内容会合并在一行，难以看清楚。

如下格式：


     
      
       aatext
       
        cctext
       
      
      
       bbtext
       
        ddtext
        
         eetext

使用minidom模块中的toprettyxml和writexml方法都有参数可以优化XML，但是有两个问题：

a. 如果解析的XML已经是美化过的，那么执行该方法会多出很多空行

b. 产生的结果会将text也独立一行，如下：

而我想产生如下结果：

于是只能自己写一个美化XML的方法。

我们首先研究一下ElementTree模块中的Element类，使用getroot方法返回的便是Element类。

该类中有四个属性tag、attrib、text与tail, 对应在XML中如下图所示：

整个XML就是一个Element，里面嵌套了很多子Element。

Element可以使用for循环迭代。

通过在text和tail中增加换行和制表符，就可以实现美化XML的目的。

美化代码如下：

def prettyXml(element, indent, newline, level = 0): # elemnt为传进来的Elment类，参数indent用于缩进，newline用于换行      if element:  # 判断element是否有子元素          if element.text == None or element.text.isspace(): # 如果element的text没有内容              element.text = newline + indent * (level + 1)            else:              element.text = newline + indent * (level + 1) + element.text.strip() + newline + indent * (level + 1)      #else:  # 此处两行如果把注释去掉，Element的text也会另起一行          #element.text = newline + indent * (level + 1) + element.text.strip() + newline + indent * level      temp = list(element) # 将elemnt转成list      for subelement in temp:          if temp.index(subelement) < (len(temp) - 1): # 如果不是list的最后一个元素，说明下一个行是同级别元素的起始，缩进应一致              subelement.tail = newline + indent * (level + 1)          else:  # 如果是list的最后一个元素， 说明下一行是母元素的结束，缩进应该少一个              subelement.tail = newline + indent * level          prettyXml(subelement, indent, newline, level = level + 1) # 对子元素进行递归操作            from xml.etree import ElementTree      #导入ElementTree模块  tree = ElementTree.parse('test.xml')   #解析test.xml这个文件，该文件内容如上文  root = tree.getroot()                  #得到根元素，Element类  prettyXml(root, '\t', '\n')            #执行美化方法  ElementTree.dump(root)                 #显示出美化后的XML内容

输出结果如下：