Python使用Mermaid与Graphviz实现关系图表生成的实战指南

pip install pydantic graphviz

sudo apt-get install graphviz

brew install graphviz

from pydantic import BaseModel, Field

class Relationship(BaseModel):
    subject: str = Field(...)
    predicate: str = Field(...)
    object: str = Field(...)

from graphviz_render import relationships_to_mermaid

# 示例关系
relationships = [
    Relationship(subject="用户", predicate="登录", object="系统"),
    Relationship(subject="系统", predicate="验证", object="身份"),
    Relationship(subject="数据库", predicate="存储", object="用户信息"),
    Relationship(subject="AI助手", predicate="处理", object="请求"),
    Relationship(subject="文档", predicate="描述", object="功能")
]

mermaid_code = relationships_to_mermaid(relationships)
print("生成的Mermaid代码:\n")
print(mermaid_code)

from pathlib import Path
from graphviz_render import relationships_to_png

output_path = Path("./test_relationships.png")

try:
    relationships_to_png(relationships, output_path)
    print(f"Graphviz图表已保存到: {output_path.absolute()}")
except Exception as e:
    print(f"生成Graphviz图表时出错: {e}")
    print("可能原因：未安装Graphviz或中文字体缺失")

dot.attr(fontname="Microsoft YaHei")

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from pydantic import BaseModel, Field
from pathlib import Path
from graphviz_render import relationships_to_mermaid, relationships_to_png

class Relationship(BaseModel):
    subject: str = Field(...)
    predicate: str = Field(...)
    object: str = Field(...)

def test_mermaid_generation():
    relationships = [
        Relationship(subject="用户", predicate="登录", object="系统"),
        Relationship(subject="系统", predicate="验证", object="身份"),
        Relationship(subject="数据库", predicate="存储", object="用户信息"),
        Relationship(subject="AI助手", predicate="处理", object="请求"),
        Relationship(subject="文档", predicate="描述", object="功能")
    ]
    mermaid_code = relationships_to_mermaid(relationships)
    print("生成的Mermaid代码:\n", mermaid_code)
    return relationships

def test_graphviz_generation(relationships):
    output_path = Path("./test_relationships.png")
    try:
        relationships_to_png(relationships, output_path)
        print(f"Graphviz图表已保存到: {output_path.absolute()}")
    except Exception as e:
        print(f"生成Graphviz图表时出错: {e}")

if __name__ == "__main__":
    rels = test_mermaid_generation()
    test_graphviz_generation(rels)