pip包依赖关系可视化

最新推荐文章于 2023-03-07 19:19:53 发布

zzy979

最新推荐文章于 2023-03-07 19:19:53 发布

阅读量660

点赞数

分类专栏： Python

本文链接：https://blog.csdn.net/zzy979481894/article/details/112385122

版权

Graphviz pip 依赖关系可视化 Python

关键词由CSDN通过智能技术生成

Python 专栏收录该内容

10 篇文章 0 订阅

订阅专栏

原创工具程序，使用Graphviz将pip包的依赖关系图可视化，效果如下图所示
效果图
其中红色表示没有入边的顶点

代码如下

import subprocess

import pip._internal.commands.list as list_cmd
import pip._internal.commands.show as show_cmd


def run():
    with open('pip_libs.dot', 'w', encoding='utf8') as f:
        f.write('digraph G {\n')
        packages = list_cmd.get_installed_distributions()
        pkg_infos = list(show_cmd.search_packages_info([p.project_name for p in packages]))
        for info in pkg_infos:
            if not info['required_by']:
                f.write('"{}" [color="red"];\n'.format(info['name']))
            else:
                f.write('"{}";\n'.format(info['name']))
        for info in pkg_infos:
            for p in info['required_by']:
                f.write('"{}" -> "{}";\n'.format(p, info['name']))
        f.write('}\n')
    try:
        subprocess.run(['dot', '-Tpng', 'pip_libs.dot', '-o', 'pip_libs.png'])
        print('关系图已输出到pip_libs.png')
    except FileNotFoundError:
        print('未安装Graphviz，关系图已输出到pip_libs.dot')


if __name__ == '__main__':
    run()