使用Python读取PlantUML导出的XMI文件

使用Python读取PlantUML导出的XMI文件

说明：

• 首次发表日期：2024-09-08
• PlantUML官网： https://plantuml.com/zh/
• PlantUML命令行文档： https://plantuml.com/zh/command-line#6a26f548831e6a8c
• PlantUML XMI文档: https://plantuml.com/zh/xmi

首先安装lxml：

pip install lxml -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple

将xmi文件解析为ElementTree:

from lxml import etree


xmi_file = "output.xmi"

tree = etree.parse(xmi_file)

获取根节点：

root = tree.getroot()

查看命名空间映射nsmap

print(root.nsmap)

{'UML': 'href://org.omg/UML/1.3'}

将命名空间保存下来备用：

UML_NAMESPACE = root.nsmap['UML']

遍历Class标签：

for element in root.iter(etree.QName(UML_NAMESPACE, 'Class')):
    print(element.get("name"))

根据需要做进一步处理，比如生成代码。