神器，轻松可视化 Python 程序调用流程

作者 | 周萝卜

今天我们来分享一个 Python 领域的神级第三方库 -- pycallgraph，通过该库并结合 graphviz 工具，就可以非常方便的完成 Python 应用程序调用流程的可视化工作~

我们先来看下效果图：

怎么样，很是惊艳吧~

下面我们就来一起完成这个可视化过程

生成图片的过程，是依赖工具 graphviz 的，我们先进行下载安装

下载地址

pipinstallpycallgraph

下面我们先写一个基础的代码

frompycallgraphimportPyCallGraph
frompycallgraph.outputimportGraphvizOutput


classBanana:
defeat(self):
pass


classPerson:
def__init__(self):
self.no_bananas()

defno_bananas(self):
self.bananas=[]

defadd_banana(self,banana):
self.bananas.append(banana)

defeat_bananas(self):
[banana.eat()forbananainself.bananas]
self.no_bananas()


defmain():
graphviz=GraphvizOutput()
graphviz.output_file='basic.png'

withPyCallGraph(output=graphviz):
person=Person()
forainrange(10):
person.add_banana(Banana())
person.eat_bananas()


if__name__=='__main__':
main()

代码比较简单，定义了两个简单类，主要 pycallgraph 的核心代码在 main 函数中，在 with 代码块下，把我们定义的代码执行一遍即可

运行上面的代码，会在当前目录下生成 basic.png 图片文件

从生成的图片可以非常清晰的看出整个代码的运行过程，从 main 代码块到各个类的初始化，可以说一目了然。

我们再来一个复杂一点的例子：

importre

frompycallgraphimportPyCallGraph
frompycallgraphimportConfig
frompycallgraph.outputimportGraphvizOutput


defmain():
graphviz=GraphvizOutput()
graphviz.output_file='regexp.png'
config=Config(include_stdlib=True)

withPyCallGraph(output=graphviz,config=config):
reo=compile()
match(reo)


defcompile():
returnre.compile('^[abetors]*$')


defmatch(reo):
[reo.match(a)forainwords()]


defwords():
return[
'abbreviation',
'abbreviations',
'abettor',
'abettors',
'abilities',
'ability',
'abrasion',
'abrasions',
'abrasive',
'abrasives',
]

if__name__=='__main__':
main()

代码同样不负责，不过在编译器内部是调用了 re 正则的，我们来看看最终生成的图片：

分享

点收藏

点点赞