python 实现xml文件的读写

1. from xml.etree.ElementTree import ElementTree,Element  
2.   
3. def read_xml(in_path):  
4.     '''''读取并解析xml文件 
5.        in_path: xml路径 
6.        return: ElementTree'''  
7.     tree = ElementTree()  
8.     tree.parse(in_path)  
9.     return tree  
10.   
11. def write_xml(tree, out_path):  
12.     '''''将xml文件写出 
13.        tree: xml树 
14.        out_path: 写出路径'''  
15.     tree.write(out_path, encoding="utf-8",xml_declaration=True)  
16.   
17. def if_match(node, kv_map):  
18.     '''''判断某个节点是否包含所有传入参数属性 
19.        node: 节点 
20.        kv_map: 属性及属性值组成的map'''  
21.     for key in kv_map:  
22.         if node.get(key) != kv_map.get(key):  
23.             return False  
24.     return True  
25.   
26. #---------------search -----  
27.   
28. def find_nodes(tree, path):  
29.     '''''查找某个路径匹配的所有节点 
30.        tree: xml树 
31.        path: 节点路径'''  
32.     return tree.findall(path)  
33.   
34.   
35. def get_node_by_keyvalue(nodelist, kv_map):  
36.     '''''根据属性及属性值定位符合的节点，返回节点 
37.        nodelist: 节点列表 
38.        kv_map: 匹配属性及属性值map'''  
39.     result_nodes = []  
40.     for node in nodelist:  
41.         if if_match(node, kv_map):  
42.             result_nodes.append(node)  
43.     return result_nodes  
44.   
45. #---------------change -----  
46.   
47. def change_node_properties(nodelist, kv_map, is_delete=False):  
48.     '''''修改/增加 /删除 节点的属性及属性值 
49.        nodelist: 节点列表 
50.        kv_map:属性及属性值map'''  
51.     for node in nodelist:  
52.         for key in kv_map:  
53.             if is_delete:   
54.                 if key in node.attrib:  
55.                     del node.attrib[key]  
56.             else:  
57.                 node.set(key, kv_map.get(key))  
58.               
59. def change_node_text(nodelist, text, is_add=False, is_delete=False):  
60.     '''''改变/增加/删除一个节点的文本 
61.        nodelist:节点列表 
62.        text : 更新后的文本'''  
63.     for node in nodelist:  
64.         if is_add:  
65.             node.text += text  
66.         elif is_delete:  
67.             node.text = ""  
68.         else:  
69.             node.text = text  
70.               
71. def create_node(tag, property_map, content):  
72.     '''''新造一个节点 
73.        tag:节点标签 
74.        property_map:属性及属性值map 
75.        content: 节点闭合标签里的文本内容 
76.        return 新节点'''  
77.     element = Element(tag, property_map)  
78.     element.text = content  
79.     return element  
80.           
81. def add_child_node(nodelist, element):  
82.     '''''给一个节点添加子节点 
83.        nodelist: 节点列表 
84.        element: 子节点'''  
85.     for node in nodelist:  
86.         node.append(element)  
87.           
88. def del_node_by_tagkeyvalue(nodelist, tag, kv_map):  
89.     '''''同过属性及属性值定位一个节点，并删除之 
90.        nodelist: 父节点列表 
91.        tag:子节点标签 
92.        kv_map: 属性及属性值列表'''  
93.     for parent_node in nodelist:  
94.         children = parent_node.getchildren()  
95.         for child in children:  
96.             if child.tag == tag and if_match(child, kv_map):  
97.                 parent_node.remove(child)  
98.                           
99.   
100.   
101. if __name__ == "__main__":  
102.       
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")  
135.       
136.