Python生成xml文件方法

一、背景介绍

由于工作中有个需求是需要生成一个jmeter脚本，而jmeter脚本其实就是一个xml文件。因此我只需要掌握如何生存一个xml文件即可

二、基本介绍

• ElementTree：表示整个XML文件
• Element：表示树中的一个元素，每个元素都有对应的标签tag
• text：获取标签内容
• attrib：获取标签的属性和属性值，是json结构
• Element.get()：访问标签的属性值
• Element.set():添加和修改标签的属性和属性值
• Element.append()：添加孩子节点
• Element.remove()：删除元素
• Element.find() ：根据标签或路径查找出第一个匹配的元素
• Element.findall()：根据标签或路径查找出所有匹配的子元素

三、示例代码

1、首先提供一个最基本的jmeter脚本模版文件。通过对该模版文件进行修改获得符合要求的jmeter压测脚本

  
    
      
      false
      true
      false
      
        
      
      
    
    
      
        continue
        
          false
          -1
        
        1
        
        true
        1
        
        false
      
      
        
          \t
          UTF-8
          press.txt
          false
          false
          false
          shareMode.all
          true
          uri,header,body
        
        
        
          
            
              ip
              
              =
            
            
              port
              
              =
            
          
        
        
        
          
        
        
        
          
            
          
          ${ip}
          ${port}
          http
          utf-8
          ${uri}
          GET
          true
          false
          true
          false
          
          
          
        
        
        
          true
          
            
              
                false
                ${body}
                =
              
            
          
          ${ip}
          ${port}
          http
          utf-8
          ${uri}
          POST
          true
          false
          true
          false
          
          
          
        
        
        
          true
          
            saveConfig
            
              true
              true
              true
              true
              true
              true
              true
              true
              true
              false
              true
              true
              false
              false
              false
              true
              false
              false
              false
              true
              0
              true
              true
              true
              true
              true
              true
            
          
          
        
        
        
          false
          
            saveConfig
            
              true
              true
              true
              true
              true
              true
              true
              true
              true
              false
              true
              true
              false
              false
              false
              true
              false
              false
              false
              true
              0
              true
              true
              true
              true
              true
              true
            
          
          
        
        
      
    
  

2、修改xml文件的python代码

tree = ET.parse(template_path)
root = tree.getroot()

for child in root.find('hashTree'):
print("curr child tag", child.tag)
if child.tag == "hashTree":
    hashTreeElement = child.find('hashTree')
    csvDataSetElement = hashTreeElement[0]
    csvDataSetElement[2].text = pressdata_file_path
    
    collectionProp = hashTreeElement[2][0]
    collectionProp[0][1].text = ip
    collectionProp[1][1].text = port

    collectionProp_of_header = hashTreeElement[4][0]
    for key in header.keys():
        new_elementProp = ET.Element('elementProp')
        new_elementProp.attrib = {"name":"", "elementType":"Header"}

        new_stringProp_name = ET.Element('stringProp')
        new_stringProp_name.text = key
        new_stringProp_name.attrib = {"name":"Header.name"}

        new_stringProp_value = ET.Element('stringProp')
        new_stringProp_value.text = header[key]
        new_stringProp_value.attrib = {"name":"Header.value"}
        
        new_elementProp.append(new_stringProp_name)
        new_elementProp.append(new_stringProp_value)
        collectionProp_of_header.append(new_elementProp)

    if method == "get":
        hashTreeElement.remove(hashTreeElement[8])
        hashTreeElement.remove(hashTreeElement[8])
    elif method == "post":
        hashTreeElement.remove(hashTreeElement[6])
        hashTreeElement.remove(hashTreeElement[6])

if child.tag == "TestPlan":
    child.set("testname", name)

代码解析：

1. 导入包 import xml.etree.ElementTree as ET
2. 使用解析器解析根元素

   # 加载XML树
   tree = ET.parse(template_file_path)
   root = tree.getroot()
   

3. 根据标签名或路径找到第一个匹配的元素

   root.find('hashTree')
   

4. 通过索引解析元素并赋值

   csvDataSetElement[2].text = pressdata_file_path
   

5. 修改标签的属性值

   child.set("testname", name)
   

6. 添加新节点元素

    new_elementProp.append(new_stringProp_value)
   

7. 删除某一个标签

   hashTreeElement.remove(hashTreeElement[8])
   

8. 创建一个新的元素

   new_stringProp_value = ET.Element('stringProp')