python生成xml文件

 1 #!/usr/bin/env python3
 2 # _*_ coding: utf-8 _*_
 3  5 from xml.dom.minidom import Document
 6 
 7 def readFile(filename, lines):
 8     with open(filename, ‘r‘) as f:
 9         for line in f:
10             line = line.rstrip(‘ \n‘)
11             if line.startswith(‘//‘) or len(line) == 0:
12                 continue
13             lines.append(line)
14 
15 def writeXml(filename, lines, tagNames):
16     # 创建doc
17     doc = Document()
18     # 创建根节点
19     root = doc.createElement(tagNames[0])
20     doc.appendChild(root)
21 
22     # 记录每层节点的最新元素
23     nodes = {0: root}
24 
25     for line in lines:
26         index = line.rfind(‘ ‘)
27         level = (index + 1) / 4 + 1
28         line = line.lstrip(‘ ‘)
29         
30         node = doc.createElement(tagNames[level])
31         node.setAttribute(‘name‘, line)
32 
33         nodes[level - 1].appendChild(node)
34         nodes[level] = node
35 
36     with open(filename, ‘w‘) as f:
37         f.write(doc.toprettyxml(indent=‘\t‘))
38 
39 def display(lines):
40     for line in lines:
41         print(line)
42 
43 if __name__ == ‘__main__‘:
44     lines = []
45     readFile(‘./file/sector.txt‘, lines)
46 
47     tagNames = [‘SectorFile‘, ‘Sectors‘, ‘sector‘, ‘sector_second‘]
48     writeXml(‘./file/sector.xml‘, lines, tagNames)