如何实现Python多模块函数调用关系的有向图可视化

存在完全适配需求的成熟Python工具库，无需从零通过正则实现调用关系识别，效率和识别准确率远高于自写正则的方案。

核心适配工具 #

• pydeps：专门用于生成Python代码调用关系图的开源工具，基于AST静态解析实现，无需运行目标代码即可完成全量扫描。支持指定多模块文件夹路径作为输入，可自动完成：
  • 收集所有自定义函数作为图的顶点集合
  • 精准识别函数内部的实际调用关系，生成从调用方指向被调用方的有向边，完全匹配需求中的边定义规则
  • 可通过参数配置过滤第三方库、内置函数的无关节点，仅保留项目自定义的函数节点，支持导出PNG、SVG、dot格式的可视化结果
• code2flow：同样基于AST静态解析的调用关系绘图工具，支持批量扫描多模块目录，可自定义顶点筛选规则，输出的有向图结构完全符合要求，适配快速生成调用关系图的场景。

为什么不推荐正则自写实现 #

• 正则匹配误判率极高：很容易将注释内容、字符串字面量里的函数名、同名变量错判为函数调用，也无法处理跨行调用、嵌套对象方法调用、条件分支内的调用等复杂场景，最终生成的调用关系图误差极大。
• 上述成熟工具的底层逻辑是直接解析代码的抽象语法树，能精准定位函数的作用域、过滤非真实调用的内容，识别准确率远高于正则匹配方案。

可自定义的轻量实现路径 #

如果需要完全自定义图的生成规则，也无需从头编写正则匹配逻辑，可基于Python生态的成熟组件快速搭建：

1. 用Python自带的ast模块遍历目标文件夹下所有Python源码，解析为语法树结构
2. 遍历语法树收集所有FunctionDef/AsyncFunctionDef类型的节点，构成图的顶点集合V
3. 遍历每个函数节点的子语法树，提取所有Call类型的真实调用节点，若被调用的函数名存在于顶点集合V中，即生成对应的有向边构成集合E
4. 用networkx存储图结构，搭配pygraphviz或matplotlib完成图的可视化

内容的提问来源于stack exchange，提问作者Veliko