如何解决如何将DOCTYPE添加到已经存在的元素树的第二行?
我有创建XML的代码,但是我需要在文件第二行使用DOCTYPE标头。我看到每个人都为此使用lxml,但是我已经在使用xml.etree.ElementTree
。我可以在同一棵树的同一文件中使用它们吗?这是我的代码:
import xml.etree.ElementTree as ET
from lxml import etree
global num_of_parallel_tests
num_of_parallel_tests = 5
def GenerateXML(fileName) :
root = ET.Element("suite")
root.set("thread-count",str(num_of_parallel_tests))
root.set("name","Selenium Automated Test w/ TestNG")
root.set("parallel","tests")
for x in range(1,num_of_parallel_tests + 1):
test = ET.Element("test")
test.set("name","Testing Row " + str(x))
parameter = ET.SubElement(test,"parameter")
parameter.set("name","Row")
parameter.set("value",str(x))
classes = ET.SubElement(test,"classes")
a_class = ET.SubElement(classes,"class")
a_class.set("name","RealPackage.ParallelTests")
root.append(test)
tree = ET.ElementTree(root)
with open (fileName,"wb") as files :
tree.write(files)
# Driver Code
if __name__ == "__main__":
GenerateXML("GeneratedXML.xml")
它会产生以下输出:
<?xml version="1.0"?>
-<suite parallel="tests" name="Selenium Automated Test w/ TestNG" thread-count="5">
-<test name="Testing Row 1">
<parameter name="Row" value="1"/>
-<classes>
<class name="RealPackage.ParallelTests"/>
</classes>
</test>
-<test name="Testing Row 2">
<parameter name="Row" value="2"/>
-<classes>
<class name="RealPackage.ParallelTests"/>
</classes>
</test>
-<test name="Testing Row 3">
<parameter name="Row" value="3"/>
-<classes>
<class name="RealPackage.ParallelTests"/>
</classes>
</test>
-<test name="Testing Row 4">
<parameter name="Row" value="4"/>
-<classes>
<class name="RealPackage.ParallelTests"/>
</classes>
</test>
-<test name="Testing Row 5">
<parameter name="Row" value="5"/>
-<classes>
<class name="RealPackage.ParallelTests"/>
</classes>
</test>
</suite>
但是我需要这个输出:
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE suite SYSTEM "https://testng.org/testng-1.0.dtd">
<suite thread-count="5" name="BlogSuite" parallel="tests">
<test name="Test 1">
<parameter name="Row" value="1"/>
<classes>
<class name="RealPackage.ParallelTests"/>
</classes>
</test>
<test name="Test 2">
<parameter name="Row" value="2"/>
<classes>
<class name="RealPackage.ParallelTests"/>
</classes>
</test>
<test name="Test 3">
<parameter name="Row" value="3"/>
<classes>
<class name="RealPackage.ParallelTests"/>
</classes>
</test>
<test name="Test 4">
<parameter name="Row" value="4"/>
<classes>
<class name="RealPackage.ParallelTests"/>
</classes>
</test>
<test name="Test 5">
<parameter name="Row" value="5"/>
<classes>
<class name="RealPackage.ParallelTests"/>
</classes>
</test>
</suite>
请注意,前两行不同。
解决方法
原来我需要的是这一行:
tree.write(files,encoding="utf-8",xml_declaration=True,doctype='<!DOCTYPE suite SYSTEM "https://testng.org/testng-1.0.dtd">')
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。