如何使用elementtree将元素添加到xml文件

时间:2013-01-21 14:07:57

标签: python xml elementtree

我有一个xml文件,我正在尝试添加其他元素。 xml具有下一个结构:

<root>
  <OldNode/>
</root>

我正在寻找的是:

<root>
  <OldNode/>
  <NewNode/>
</root>

但实际上我正在接下来的xml:

<root>
  <OldNode/>
</root>

<root>
  <OldNode/>
  <NewNode/>
</root>

我的代码看起来像这样:

file = open("/tmp/" + executionID +".xml", 'a')
xmlRoot = xml.parse("/tmp/" + executionID +".xml").getroot()

child = xml.Element("NewNode")
xmlRoot.append(child)

xml.ElementTree(root).write(file)

file.close()

感谢。

1 个答案:

答案 0 :(得分:8)

您打开了要追加的文件,这会将数据添加到最后。使用w模式打开文件进行编写。更好的是,只需在ElementTree对象上使用.write()方法:

tree = xml.parse("/tmp/" + executionID +".xml")

xmlRoot = tree.getroot()
child = xml.Element("NewNode")
xmlRoot.append(child)

tree.write("/tmp/" + executionID +".xml")

使用.write()方法的另一个好处是,您可以设置编码,强制在需要时编写XML序言等。

如果您必须使用打开的文件来美化XML,请使用'w'模式,'a'打开一个文件进行追加,从而导致您观察到的行为:< / p>

with open("/tmp/" + executionID +".xml", 'w') as output:
     output.write(prettify(tree))

其中prettify类似于:

from xml.etree import ElementTree
from xml.dom import minidom

def prettify(elem):
    """Return a pretty-printed XML string for the Element.
    """
    rough_string = ElementTree.tostring(elem, 'utf-8')
    reparsed = minidom.parseString(rough_string)
    return reparsed.toprettyxml(indent="  ")

e.g。 minidom美化技巧。