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(' \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的字典，用於存放當前層級的最新節點。便於字節點查找父節點