先上代码:
#!/usr/bin/env python3
# _*_ coding: utf-8 _*_
from xml.dom.minidom import Document def readFile(filename, lines):
with open(filename, 'r') as f:
for line in f:
line = line.rstrip(' \n')
if line.startswith('//') or len(line) == 0:
continue
lines.append(line) def writeXml(filename, lines, tagNames):
# 创建doc
doc = Document()
# 创建根节点
root = doc.createElement(tagNames[0])
doc.appendChild(root) # 记录每层节点的最新元素
nodes = {0: root} for line in lines:
index = line.rfind(' ')
level = (index + 1) / 4 + 1
line = line.lstrip(' ') node = doc.createElement(tagNames[level])
node.setAttribute('name', line) nodes[level - 1].appendChild(node)
nodes[level] = node with open(filename, 'w') as f:
f.write(doc.toprettyxml(indent='\t')) def display(lines):
for line in lines:
print(line) if __name__ == '__main__':
lines = []
readFile('./file/sector.txt', lines) tagNames = ['SectorFile', 'Sectors', 'sector', 'sector_second']
writeXml('./file/sector.xml', lines, tagNames)
1. 使用库 xml.dom.minidom
2. readFile函数
这个函数的功能上读取文件,并把每一行内容除去右边空格,存入列表中
参数一:文件名
参数二:文件行存入的列表
3. writeXml函数
功能:读取列表,生成xml文件
参数一:生成的xml文件名
参数二:文件行列表
参数三:xml标签名
此处读取的文件,左侧是通过tab键进行分层的。但由于编辑器设置的一个tab键,替换成四个空格。所以此处的规律是查找最后一个空格出现的位置,加1除4就可以得到该文件在xml中所处的层级。
第22行:此处的目的是建立一个level-node的字典,用于存放当前层级的最新节点。便于字节点查找父节点