如何使用python的xml庫實(shí)現(xiàn)自閉合標(biāo)簽

更新時(shí)間：2025年01月07日 10:00:59 作者：knighthood2001

文章介紹了作者編寫一個(gè)URDF格式化插件的初衷,目的是解決sw2urdf導(dǎo)出的URDF文件格式混亂的問題,本文結(jié)合實(shí)例代碼給大家介紹的非常詳細(xì),感興趣的朋友一起看看吧

前言

最近一段時(shí)間一直想要寫一個(gè)urdf格式化插件。

至于為什么嘛，因?yàn)槭褂胹w2urdf插件，導(dǎo)出的urdf，同一標(biāo)簽的內(nèi)容，是跨行的，這就導(dǎo)致，內(nèi)容比較亂，而且行數(shù)比較多。影響閱讀。

因此，自己想寫格式化的腳本。

最近就開始分享一些，之前的思考。

正文

import xml.etree.ElementTree as ET
def pretty_print(element, level=0, indent="  "):
    """
    手動(dòng)格式化 XML 數(shù)據(jù)，添加縮進(jìn)并將空元素轉(zhuǎn)換為自閉合標(biāo)簽。
    :param element: XML 元素
    :param level: 當(dāng)前縮進(jìn)級(jí)別
    :param indent: 縮進(jìn)字符（默認(rèn)為兩個(gè)空格）
    :return: 格式化后的 XML 字符串
    """
    result = ""
    if len(element) == 0:  # 如果沒有子元素，即可能是空元素
        if element.text and element.text.strip():  # 如果有文本內(nèi)容
            result += f"{indent * level}<{element.tag}>{element.text.strip()}</{element.tag}>\n"
        else:
            result += f"{indent * level}<{element.tag} "
            # 如果有屬性，添加到標(biāo)簽中
            if element.attrib:
                result += " ".join([f'{key}="{value}"' for key, value in element.attrib.items()])
            result += " />\n"  # 使用自閉合標(biāo)簽形式
    else:
        result += f"{indent * level}<{element.tag}"
        # 如果有屬性，添加到標(biāo)簽中
        if element.attrib:
            result += " " + " ".join([f'{key}="{value}"' for key, value in element.attrib.items()])
        result += ">\n"
        # 處理元素的文本內(nèi)容
        if element.text and element.text.strip():
            result += f"{indent * (level + 1)}{element.text.strip()}\n"
        # 遞歸處理子元素
        for child in element:
            result += pretty_print(child, level + 1, indent)
        # 處理結(jié)束標(biāo)簽
        result += f"{indent * level}</{element.tag}>\n"
    return result
# 讀取 XML 文件
file_path = 'test.urdf'  # 請(qǐng)將此路徑替換為實(shí)際的文件路徑
tree = ET.parse(file_path)
root = tree.getroot()
# 直接格式化 root 元素，而不需要?jiǎng)?chuàng)建新的根節(jié)點(diǎn)
formatted_xml = pretty_print(root)
# 打印格式化后的 XML
print(formatted_xml)
with open('test_pretty.urdf', 'w') as f:
    f.write(formatted_xml)

這里，代碼主要是通過xml.etree.ElementTree解析xml文件。

然后在函數(shù)中補(bǔ)充上空格，<，/>等標(biāo)簽。

注意，這個(gè)函數(shù)是個(gè)遞歸的函數(shù)，會(huì)在函數(shù)內(nèi)部調(diào)用函數(shù)本身。

實(shí)現(xiàn)的效果如下

可以發(fā)現(xiàn)，目前的問題是：

沒有注釋
urdf最開始的標(biāo)簽，不見了。

到此這篇關(guān)于使用python的xml庫實(shí)現(xiàn)自閉合標(biāo)簽的文章就介紹到這了,更多相關(guān)python xml庫自閉合標(biāo)簽內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: