用ElementTree在一个xml文件中写入多个xml结构

Question

我正在尝试用ElementTree编写一个xml文件。GeeksforGeeks有一个很好的教程。我面临的问题是，我想在同一个xml文件中编写多个xml结构。

在本教程期间创建的xml文件如下：

电流输出

<chess>
    <Opening>
        <E4 type="Accepted">King's Gambit Accepted</E4>
        <D4 type="Declined">Queen's Gambit Declined</D4>
    </Opening>
</chess>

我想用循环创建以下内容：

期望输出

<chess>
    <Opening>
        <E4 type="Accepted">King's Gambit Accepted</E4>
        <D4 type="Declined">Queen's Gambit Declined</D4>
    </Opening>
</chess>
<chess>
    <Opening>
        <E4 type="Accepted">King's Gambit Accepted</E4>
        <D4 type="Declined">Queen's Gambit Declined</D4>
    </Opening>
</chess>
<chess>
    <Opening>
        <E4 type="Accepted">King's Gambit Accepted</E4>
        <D4 type="Declined">Queen's Gambit Declined</D4>
    </Opening>
</chess>

我尝试使用循环函数，循环正在工作，但是对xml文件的写入不起作用。我的代码看起来是这样的。

码

import xml.etree.ElementTree as ET
test = [1,2,3]
l = []

for i in test:
    data = ET.Element('chess')
    element1 = ET.SubElement(data, 'Opening')
    s_elem1 = ET.SubElement(element1, 'E4')
    s_elem2 = ET.SubElement(element1, 'D4')
    s_elem1.set('type', 'Accepted')
    s_elem2.set('type', 'Declined')
    s_elem1.text = "King's Gambit Accepted"
    s_elem2.text = "Queen's Gambit Declined"
    b_xml = ET.tostring(data)
    l.append(b_xml)

output = bytearray(l)

with open("output.xml", "wb") as f:
    f.write(l)

错误消息

TypeError: an integer is required

是否有办法在一个xml文件中创建多个结构并将其写入输出文件？

Answer 1

您需要在XML中至少有一个根元素，才能将子元素追加到：

from lxml import etree
test = [1, 2, 3]
l = []

xmlRoot = etree.Element('root')

for i in test:
    data = etree.SubElement(xmlRoot, 'chess')
    element1 = etree.SubElement(data, 'Opening')
    s_elem1 = etree.SubElement(element1, 'E4')
    s_elem2 = etree.SubElement(element1, 'D4')
    s_elem1.set('type', 'Accepted')
    s_elem2.set('type', 'Declined')
    s_elem1.text = "King's Gambit Accepted"
    s_elem2.text = "Queen's Gambit Declined"

doc = etree.ElementTree(xmlRoot) # convert into elementtree and write it directly into a file
with open("output.xml", "wb") as f:
    f.write(
        etree.tostring(
            doc, pretty_print=True, xml_declaration=True, encoding='utf-8'
        )
    )

print(open("output.xml").read())

退出：

<?xml version='1.0' encoding='utf-8'?>
<root>
  <chess>
    <Opening>
      <E4 type="Accepted">King's Gambit Accepted</E4>
      <D4 type="Declined">Queen's Gambit Declined</D4>
    </Opening>
  </chess>
  <chess>
    <Opening>
      <E4 type="Accepted">King's Gambit Accepted</E4>
      <D4 type="Declined">Queen's Gambit Declined</D4>
    </Opening>
  </chess>
  <chess>
    <Opening>
      <E4 type="Accepted">King's Gambit Accepted</E4>
      <D4 type="Declined">Queen's Gambit Declined</D4>
    </Opening>
  </chess>
</root>