python实现合并多个dot文件

前言：通过Joern生成源程序的cpg图为多个零散的dot文件，对其进行分析或可视化之前需合并，以下为利用pygraphviz库对dot文件进行合并的python代码实现（pygraphviz库的安装方式见：正确安装pygraphviz库-CSDN博客）：

import pygraphviz as pgvdef merge_dot_files_pygraphviz(input_files, output_file):# 创建一个空的有向图merged_graph = pgv.AGraph(directed=True)for file_path in input_files:g = pgv.AGraph(file_path)for node in g.nodes():merged_graph.add_node(node, **g.get_node(node).attr)for edge in g.edges():merged_graph.add_edge(edge[0], edge[1], **g.get_edge(edge[0], edge[1]).attr)# 保存合并后的图merged_graph.write(output_file)print(merged_graph)print(f"合并完成，输出文件为：{output_file}")# 示例用法
if __name__ == "__main__":input_files = ["0-cpg.dot", "1-cpg.dot", "2-cpg.dot"]output_file = "merged_graph.dot"merge_dot_files_pygraphviz(input_files, output_file)