python生成xml文件

先上代码：

#!/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(' 
')
            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='	'))

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的字典，用于存放当前层级的最新节点。便于字节点查找父节点