在Python eTree中美化xml

时间:2015-06-22 19:26:18

标签: python xml lxml elementtree

我有这个简单的py脚本生成一个xml文件并保存,它并且想知道是否有一种简单的方法来缩进它?

\S

我查看了其他一些SO Q& A的Pretty printing XML in Python,但这些似乎主要需要其他外部库?并且想知道是否有办法不使用那些?

感谢您的帮助。

1 个答案:

答案 0 :(得分:0)

您可以使用标准库的communicate模块minidom

import xml.dom.minidom as minidom

xml = minidom.Document()
root = xml.createElement("root")
xml.appendChild(root)

doc = xml.createElement("doc")
doc.setAttribute("location", "one")
root.appendChild(doc)

field = xml.createElement("field1")
field.setAttribute("name", "blah")
text = xml.createTextNode("some value1")
field.appendChild(text)
doc.appendChild(field)

field = xml.createElement("field2")
field.setAttribute("name", "asdfasd")
text = xml.createTextNode("some value2")
field.appendChild(text)

doc.appendChild(field)
print(xml.toprettyxml(indent=' '*4))

产量

<?xml version="1.0" ?>
<root>
    <doc location="one">
        <field1 name="blah">some value1</field1>
        <field2 name="asdfasd">some value2</field2>
    </doc>
</root>

或者,如果您更喜欢ElementTree方法来创建XML而不介意 由于效率有点低,您可以使用ElementTree来编写未格式化的XML 到toprettyxml method(对于Python2)或StringIO(对于Python3),将其解析为minidom 记录,然后使用toprettyxml再次将其写回:

import xml.etree.cElementTree as ET
import xml.dom.minidom as minidom

try:
    # for Python2
    from cStringIO import StringIO as BytesIO
except ImportError:
    # for Python3
    from io import BytesIO

root = ET.Element("root")
doc = ET.SubElement(root, "doc", location="one")

ET.SubElement(doc, "field1", name="blah").text = "some value1"
ET.SubElement(doc, "field2", name="asdfasd").text = "some vlaue2"
buf = BytesIO()
buf.write(ET.tostring(root))
buf.seek(0)
root = minidom.parse(buf)
print(root.toprettyxml(indent=' '*4))