神器，輕松可視化 Python 程序調(diào)用流程
作者：周蘿卜
今天我們來分享一個(gè) Python 領(lǐng)域的神級(jí)第三方庫 -- pycallgraph，通過該庫并結(jié)合 graphviz 工具，就可以非常方便的完成 Python 應(yīng)用程序調(diào)用流程的可視化工作。
我們先來看下效果圖：
怎么樣，很是驚艷吧~
下面我們就來一起完成這個(gè)可視化過程。
1. 安裝 graphviz 工具
生成圖片的過程，是依賴工具 graphviz 的，我們先進(jìn)行下載安裝。
下載地址
- http://www.graphviz.org/download/
2. 實(shí)戰(zhàn)
接下來我們還需要安裝兩個(gè) Python 依賴庫。
pip install pycallgraph
下面我們先寫一個(gè)基礎(chǔ)的代碼;
from pycallgraph import PyCallGraph
from pycallgraph.output import GraphvizOutput
class Banana:
def eat(self):
pass
class Person:
def __init__(self):
self.no_bananas()
def no_bananas(self):
self.bananas = []
def add_banana(self, banana):
self.bananas.append(banana)
def eat_bananas(self):
[banana.eat() for banana in self.bananas]
self.no_bananas()
def main():
graphviz = GraphvizOutput()
graphviz.output_file = 'basic.png'
with PyCallGraph(output=graphviz):
person = Person()
for a in range(10):
person.add_banana(Banana())
person.eat_bananas()
if __name__ == '__main__':
main()
代碼比較簡單，定義了兩個(gè)簡單類，主要 pycallgraph 的核心代碼在 main 函數(shù)中，在 with 代碼塊下，把我們定義的代碼執(zhí)行一遍即可
運(yùn)行上面的代碼，會(huì)在當(dāng)前目錄下生成 basic.png 圖片文件
從生成的圖片可以非常清晰的看出整個(gè)代碼的運(yùn)行過程，從 main 代碼塊到各個(gè)類的初始化，可以說一目了然。
我們?cè)賮硪粋€(gè)復(fù)雜一點(diǎn)的例子：
import re
from pycallgraph import PyCallGraph
from pycallgraph import Config
from pycallgraph.output import GraphvizOutput
def main():
graphviz = GraphvizOutput()
graphviz.output_file = 'regexp.png'
config = Config(include_stdlib=True)
with PyCallGraph(output=graphviz, config=config):
reo = compile()
match(reo)
def compile():
return re.compile('^[abetors]*$')
def match(reo):
[reo.match(a) for a in words()]
def words():
return [
'abbreviation',
'abbreviations',
'abettor',
'abettors',
'abilities',
'ability',
'abrasion',
'abrasions',
'abrasive',
'abrasives',
]
if __name__ == '__main__':
main()
代碼同樣不負(fù)責(zé)，不過在編譯器內(nèi)部是調(diào)用了 re 正則的，我們來看看最終生成的圖片：
可以看到整個(gè)代碼過程復(fù)雜了很多，因?yàn)閮?nèi)部調(diào)用了很多正則內(nèi)部函數(shù)等，但是整體還是非常清晰的
可以說這個(gè)神級(jí)第三方庫，絕對(duì)是眾多 Python 愛好者，尤其是剛剛?cè)腴T Python 領(lǐng)域的朋友的福音，當(dāng)我們遇到某些不熟悉的較為復(fù)雜的代碼塊時(shí)，不妨使用該庫來嘗試一下可視化，看看能不能從中爆發(fā)靈感呢~
責(zé)任編輯：龐桂玉
來源：
小詹學(xué)Python
