python xml.etree.ElementTree附加到子元素

Question

我正在嘗試使用xml.etree.ElementTree來解析xml文件，查找特定標記，將子項附加到該標記，將另一個子項附加到新創建的標記並將文本添加到后一個子項。

我的XML：

<root>
<a>
    <b>
      <c>text1</c>
    </b>
    <b>
      <c>text2</c>
   </b>
</a>
</root>    

期望的XML：

<root>
<a>
    <b>
      <c>text1</c>
    </b>
    <b>
      <c>text2</c>
   </b>
    <b>
      <c>text3</c>
   </b>
</a>
</root>

當前代碼：

import xml.etree.ElementTree as ET
tree = ET.parse('test.xml')
root = tree.getroot()


for x in root.iter():
    if (x.tag == 'a'):
        ET.SubElement(x, 'b')
        ET.SubElement(x, 'c')
        #add text

這似乎有效，除了'c'作為'a'而不是'b'的孩子。

像這樣：

<root>
<a>
    <b>
      <c>test1</c>
    </b>
    <b>
      <c>test2</c>
    </b>
  <b /><c/></a>
</root>

另外，如何將文本添加到新創建的元素“c”中？ 我可以迭代直到找到沒有文字的標簽'c'，但必須有更好的方法。

Answer 1

您需要將b指定為c的父元素。

另外，為了獲得a標簽你不需要循環 - 只需取根（ a ）。

import xml.etree.ElementTree as ET

tree = ET.parse('test.xml')
root = tree.getroot()

a = root.find('a')
b = ET.SubElement(a, 'b')
c = ET.SubElement(b, 'c')
c.text = 'text3'

print ET.tostring(root)

打印：

<root>
    <a>
        <b>
          <c>text1</c>
        </b>
        <b>
          <c>text2</c>
        </b>
        <b>
          <c>text3</c>
        </b>
    </a>
</root>

Answer 2

我更喜歡定義自己添加文本的功能：

def SubElementWithText(parent, tag, text):
    attrib = {}
    element = parent.makeelement(tag, attrib)
    parent.append(element)
    element.text = text
    return element

這樣使用起來非常方便：

import xml.etree.ElementTree as ET

tree = ET.parse('test.xml')
root = tree.getroot()

a = root.find('a')
b = ET.SubElement(a, 'b')
c = SubElementWithText(b, 'c', 'text3')