作者 | 周蘿卜

今天我們來分享一個(gè) Python 領(lǐng)域的神級(jí)第三方庫 -- pycallgraph，通過該庫并結(jié)合 graphviz 工具，就可以非常方便的完成 Python 應(yīng)用程序調(diào)用流程的可視化工作~

我們先來看下效果圖：

怎么樣，很是驚艷吧~

下面我們就來一起完成這個(gè)可視化過程

安裝 graphviz 工具

生成圖片的過程，是依賴工具 graphviz 的，我們先進(jìn)行下載安裝

下載地址

http://www.graphviz.org/download/

實(shí)戰(zhàn)

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 正則的，我們來看看最終生成的圖片：

推薦閱讀

牛逼！Python常用數(shù)據(jù)類型的基本操作（長文系列第①篇）
牛逼！Python的判斷、循環(huán)和各種表達(dá)式（長文系列第②篇）
牛逼！Python函數(shù)和文件操作（長文系列第③篇）
牛逼！Python錯(cuò)誤、異常和模塊（長文系列第④篇）

吳恩達(dá)deeplearining.ai的經(jīng)典總結(jié)資料

Ps：從小程序直接獲取下載