mysql导出树状结构图

import mysql.connector
from graphviz import Digraph

# 连接MySQL数据库
db = mysql.connector.connect(
    host="localhost",
    user="your_username",
    password="your_password",
    database="your_database"
)
cursor = db.cursor()

# 查询树状结构数据
cursor.execute("SELECT id, parent_id, name FROM your_table")
rows = cursor.fetchall()

# 构建树状结构
dot = Digraph(comment='Tree Structure')
nodes = set()
edges = []

for row in rows:
    node_id, parent_id, name = row
    nodes.add(node_id)
    if parent_id is not None:
        edges.append((parent_id, node_id))

for node in nodes:
    dot.node(str(node), name)

for edge in edges:
    dot.edge(str(edge[0]), str(edge[1]))

# 保存图形
dot.render('tree_structure.gv', view=True)

# 关闭数据库连接
cursor.close()
db.close()