1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
#!/usr/bin/env python
"""
This example demonstrates the use of filtering.
"""
import time
from pycallgraph2 import PyCallGraph
from pycallgraph2 import Config
from pycallgraph2 import GlobbingFilter
from pycallgraph2.output import GraphvizOutput
class Banana:
def __init__(self):
pass
def eat(self):
self.secret_function()
self.chew()
self.swallow()
def secret_function(self):
time.sleep(0.2)
def chew(self):
pass
def swallow(self):
pass
def run(name, trace_filter=None, config=None, comment=None):
if not config:
config = Config()
if trace_filter:
config.trace_filter = trace_filter
graphviz = GraphvizOutput()
graphviz.output_file = 'filter-{}.png'.format(name)
if comment:
graphviz.graph_attributes['graph']['label'] = comment
with PyCallGraph(config=config, output=graphviz):
banana = Banana()
banana.eat()
def filter_none():
run(
'none',
comment='Default filtering.'
)
def filter_exclude():
trace_filter = GlobbingFilter(exclude=[
'pycallgraph.*',
'*.secret_function',
])
run(
'exclude',
trace_filter=trace_filter,
comment='Should not include secret_function.',
)
def filter_include():
trace_filter = GlobbingFilter(include=[
'*.secret_function',
'Banana.eat',
])
run(
'include',
trace_filter=trace_filter,
comment='Should show secret_function.'
)
def filter_depth():
config = Config()
config.max_depth = 1
run(
'max_depth',
config=config,
comment='Should only show a depth of one.'
)
def filter_pycallgraph():
trace_filter = GlobbingFilter(exclude=[])
run(
'pycallgraph',
trace_filter=trace_filter,
comment="Don't filter pycallgraph calls.",
)
def main():
filter_none()
filter_exclude()
filter_include()
filter_depth()
filter_pycallgraph()
if __name__ == '__main__':
main()
|
