python 生成xml文件,以及美化的實(shí)例代碼

# -*- coding:utf-8 -*- 
import os
import json
import numpy as np 
#from xml.etree import ElementTree as  etree
from xml.etree.ElementTree import Element
from xml.etree.ElementTree import SubElement
from xml.etree.ElementTree import ElementTree
 
imagePath = r'E:\Desktop\SteelCoilsDetection\test\images'
jsonPath = r'E:\Desktop\SteelCoilsDetection\test\json'
savePath = r'E:\Desktop\SteelCoilsDetection\test\xml'
 
jsonList = os.listdir(jsonPath)
for jsonName in jsonList:
    print(jsonName)
    readPath = os.path.join(jsonPath, jsonName)
 
    # 打開(kāi)json文件
    with open(readPath, 'r') as file_loader:
        jsonDic = json.load(file_loader)
 
    # print(jsonDic.keys())
    # dict_keys(['version', 'flags', 'shapes', 'imagePath', 'imageData', 'imageHeight', 'imageWidth'])
    
    # 生成xml文件
    annotation = Element('annotation')
 
    folder = SubElement(annotation, 'folder')
    folder.text = "images"
 
    filename = SubElement(annotation, 'filename')
    filename.text = jsonName.split('.')[0]
 
    path = SubElement(annotation, 'path')
    path.text = imagePath + jsonName.split('.')[0]
 
    source = SubElement(annotation, 'source')
    database = SubElement(source, 'database')
    database.text = "Unknown"
 
    size = SubElement(annotation, 'size')
    width = SubElement(size, 'width')
    width.text = str(jsonDic['imageWidth'])
    height = SubElement(size, 'height')
    height.text = str(jsonDic['imageHeight'])
    depth = SubElement(size, 'depth')
    depth.text = "3"
 
    segmented = SubElement(annotation, 'segmented')
    segmented.text = "0" 
    for shape in jsonDic['shapes']: 
        if shape["label"] == 'a':
            continue        
 
        object = SubElement(annotation, 'object') 
        name = SubElement(object, 'name')
        name.text = shape["label"]
 
        pose = SubElement(object, 'pose')
        pose.text = 'Unspecified'
 
        truncated = SubElement(object, 'truncated')
        truncated.text = str(0)
 
        difficult = SubElement(object, 'difficult')
        difficult.text = str(0)        
 
        points = shape['points']
        mritx = np.array(points)
        xxmin = min(mritx[:, 0])
        xxmax = max(mritx[:, 0])
        yymin = min(mritx[:, 1])
        yymax = max(mritx[:, 1])
 
        bndbox = SubElement(object, 'bndbox')
        xmin = SubElement(bndbox, 'xmin')
        xmin.text = str(int(xxmin))
        ymin = SubElement(bndbox, 'ymin')
        ymin.text = str(int(yymin))
        xmax = SubElement(bndbox, 'xmax')
        xmax.text = str(int(xxmax))
        ymax = SubElement(bndbox, 'ymax')
        ymax.text = str(int(yymax)) 
    tree = ElementTree(annotation)
    tree.write(os.path.join(savePath, jsonName.split('.')[0]+'.xml'), encoding = 'utf-8')

# -*- coding:utf-8 -*- 
import os
from xml.etree import ElementTree      # 導(dǎo)入ElementTree模塊  
# elemnt為傳進(jìn)來(lái)的Elment類(lèi)，參數(shù)indent用于縮進(jìn)，newline用于換行
def prettyXml(element, indent, newline, level = 0):
 
    # 判斷element是否有子元素
    if element:
 
        # 如果element的text沒(méi)有內(nèi)容
        if element.text == None or element.text.isspace():
            element.text = newline + indent * (level + 1)
        else:
            element.text = newline + indent * (level + 1) + element.text.strip() + newline + indent * (level + 1)
 
    # 此處兩行如果把注釋去掉，Element的text也會(huì)另起一行 
    #else:
        #element.text = newline + indent * (level + 1) + element.text.strip() + newline + indent * level
 
    temp = list(element) # 將elemnt轉(zhuǎn)成list
    for subelement in temp:
        # 如果不是list的最后一個(gè)元素，說(shuō)明下一個(gè)行是同級(jí)別元素的起始，縮進(jìn)應(yīng)一致
        if temp.index(subelement)  (len(temp) - 1):
            subelement.tail = newline + indent * (level + 1)
        else:  # 如果是list的最后一個(gè)元素， 說(shuō)明下一行是母元素的結(jié)束，縮進(jìn)應(yīng)該少一個(gè)
            subelement.tail = newline + indent * level   
 
        # 對(duì)子元素進(jìn)行遞歸操作 
        prettyXml(subelement, indent, newline, level = level + 1)  
 
dir = r'E:\Desktop\SteelCoilsDetection\test\xml'
for fileName in os.listdir(dir):
    print(fileName)
    tree = ElementTree.parse(os.path.join(dir, fileName))   #解析test.xml這個(gè)文件，該文件內(nèi)容如上文
    root = tree.getroot()                                   #得到根元素，Element類(lèi)
    prettyXml(root, '\t', '\n')                             # 執(zhí)行美化方法
 
    #ElementTree.dump(root)                                 #顯示出美化后的XML內(nèi)容
    tree.write(os.path.join(dir, fileName), encoding = 'utf-8')

?xml version="1.0"?>
bookstore>
	book name="西游記">
		author>吳承恩/author>
		dynasty>明朝/dynasty>
		similar name="封神演義" author="許仲琳"/>
	/book>
	book name="紅樓夢(mèng)">
		author>曹雪芹/author>
		dynasty>清朝/dynasty>
	/book>
	book name="三國(guó)演義">
		author>羅貫中/author>
		dynasty>明末清初/dynasty>
		similar name="三國(guó)志" author="陳壽"/>
	/book>
/bookstore>

import xml.etree.ElementTree as ET
tree = ET.parse("./book.xml")
root = tree.getroot()

with open("./book.xml") as fp:
    root = ET.fromstring(fp.read())

# 遍歷直接子節(jié)點(diǎn)
for book in root:
    print(book.tag, book.attrib, book.get("name"))
# 訪問(wèn)根節(jié)點(diǎn)下的第2個(gè)子節(jié)點(diǎn), 再向下訪問(wèn)第1個(gè)子節(jié)點(diǎn)的文本, 也就是 "author>曹雪芹/author>"
author = root[1][0].text
print(type(author), author)

# 遞歸選擇所有標(biāo)簽名為 "similar" 的節(jié)點(diǎn)
for book in root.iter("similar"):
    print(book.attrib)