python修改xml文件标签的属性值

需求
在实际应用中，需要对xml配置文件进行实时修改，

1.增加、删除 某些节点
2.增加，删除，修改某个节点下的某些属性
3.增加，删除，修改某些节点的文本

参考官网：xml.etree.ElementTree --- ElementTree XML API — Python 3.12.0a1 文档

使用xml文档

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
<?xml version="1.0" encoding="UTF-8"?>
<framework>
  <processers>
    <processer name="AProcesser" file="lib64/A.so" path="/tmp"/>
    <processer name="BProcesser" file="lib64/B.so" value="fordelete"/>
    <processer name="BProcesser" file="lib64/B.so2222222"/>
    <services>
      <service name="search" prefix="/bin/search?" output_formatter="OutPutFormatter:service_inc">
        <chain sequency="chain1"/>
        <chain sequency="chain2"/>
      </service>
      <service name="update" prefix="/bin/update?">
        <chain sequency="chain3" value="fordelete"/>
      </service>
    </services>
  </processers>
</framework>

　　

实现思想
使用ElementTree，先将文件读入，解析成树，之后，根据路径，可以定位到树的每个节点，再对节点进行修改，最后直接将其输出

实现代码

  1 #!/usr/bin/python
  2 # -*- coding=utf-8 -*-
  3 # author : wklken@yeah.net
  4 # date: 2012-05-25
  5 # version: 0.1
  6  
  7 from xml.etree.ElementTree import ElementTree,Element
  8  
  9 def read_xml(in_path):
 10   '''读取并解析xml文件
 11     in_path: xml路径
 12     return: ElementTree'''
 13   tree = ElementTree()
 14   tree.parse(in_path)
 15   return tree
 16  
 17 def write_xml(tree, out_path):
 18   '''将xml文件写出
 19     tree: xml树
 20     out_path: 写出路径'''
 21   tree.write(out_path, encoding="utf-8",xml_declaration=True)
 22  
 23 def if_match(node, kv_map):
 24   '''判断某个节点是否包含所有传入参数属性
 25     node: 节点
 26     kv_map: 属性及属性值组成的map'''
 27   for key in kv_map:
 28     if node.get(key) != kv_map.get(key):
 29       return False
 30   return True
 31  
 32 #---------------search -----
 33 def find_nodes(tree, path):
 34   '''查找某个路径匹配的所有节点
 35     tree: xml树
 36     path: 节点路径'''
 37   return tree.findall(path)
 38  
 39 def get_node_by_keyvalue(nodelist, kv_map):
 40   '''根据属性及属性值定位符合的节点，返回节点
 41     nodelist: 节点列表
 42     kv_map: 匹配属性及属性值map'''
 43   result_nodes = []
 44   for node in nodelist:
 45     if if_match(node, kv_map):
 46       result_nodes.append(node)
 47   return result_nodes
 48  
 49 #---------------change -----
 50 def change_node_properties(nodelist, kv_map, is_delete=False):
 51   '''修改/增加 /删除 节点的属性及属性值
 52     nodelist: 节点列表
 53     kv_map:属性及属性值map'''
 54   for node in nodelist:
 55     for key in kv_map:
 56       if is_delete:
 57         if key in node.attrib:
 58           del node.attrib[key]
 59       else:
 60         node.set(key, kv_map.get(key))
 61  
 62 def change_node_text(nodelist, text, is_add=False, is_delete=False):
 63   '''改变/增加/删除一个节点的文本
 64     nodelist:节点列表
 65     text : 更新后的文本'''
 66   for node in nodelist:
 67     if is_add:
 68       node.text += text
 69     elif is_delete:
 70       node.text = ""
 71     else:
 72       node.text = text
 73  
 74 def create_node(tag, property_map, content):
 75   '''新造一个节点
 76     tag:节点标签
 77     property_map:属性及属性值map
 78     content: 节点闭合标签里的文本内容
 79     return 新节点'''
 80   element = Element(tag, property_map)
 81   element.text = content
 82   return element
 83  
 84 def add_child_node(nodelist, element):
 85   '''给一个节点添加子节点
 86     nodelist: 节点列表
 87     element: 子节点'''
 88   for node in nodelist:
 89     node.append(element)
 90  
 91 def del_node_by_tagkeyvalue(nodelist, tag, kv_map):
 92   '''同过属性及属性值定位一个节点，并删除之
 93     nodelist: 父节点列表
 94     tag:子节点标签
 95     kv_map: 属性及属性值列表'''
 96   for parent_node in nodelist:
 97     children = parent_node.getchildren()
 98     for child in children:
 99       if child.tag == tag and if_match(child, kv_map):
100         parent_node.remove(child)
101  
102 if __name__ == "__main__":
103   #1. 读取xml文件
104   tree = read_xml("./test.xml")
105  
106   #2. 属性修改
107    #A. 找到父节点
108   nodes = find_nodes(tree, "processers/processer")
109    #B. 通过属性准确定位子节点
110   result_nodes = get_node_by_keyvalue(nodes, {"name":"BProcesser"})
111    #C. 修改节点属性
112   change_node_properties(result_nodes, {"age": "1"})
113    #D. 删除节点属性
114   change_node_properties(result_nodes, {"value":""}, True)
115  
116   #3. 节点修改
117    #A.新建节点
118   a = create_node("person", {"age":"15","money":"200000"}, "this is the firest content")
119    #B.插入到父节点之下
120   add_child_node(result_nodes, a)
121  
122   #4. 删除节点
123     #定位父节点
124   del_parent_nodes = find_nodes(tree, "processers/services/service")
125     #准确定位子节点并删除之
126   target_del_node = del_node_by_tagkeyvalue(del_parent_nodes, "chain", {"sequency" : "chain1"})
127  
128   #5. 修改节点文本
129     #定位节点
130   text_nodes = get_node_by_keyvalue(find_nodes(tree, "processers/services/service/chain"), {"sequency":"chain3"})
131   change_node_text(text_nodes, "new text")
132  
133   #6. 输出到结果文件
134   write_xml(tree, "./out.xml")

 

posted @ 2022-11-01 14:03  Hear7  阅读(620)  评论(0)  编辑  收藏  举报