-
Notifications
You must be signed in to change notification settings - Fork 0
/
yappi.py
69 lines (55 loc) · 1.69 KB
/
yappi.py
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
'''
yappi.py
Yet Another Python Profiler
Sumer Cip 2010
'''
import sys
import threading
import _yappi
__all__ = ['start', 'stop', 'enum_stats', 'print_stats', 'clear_stats']
SORTTYPE_NAME = _yappi.SORTTYPE_NAME
SORTTYPE_NCALL = _yappi.SORTTYPE_NCALL
SORTTYPE_TTOTAL = _yappi.SORTTYPE_TTOTAL
SORTTYPE_TSUB = _yappi.SORTTYPE_TSUB
SORTTYPE_TAVG = _yappi.SORTTYPE_TAVG
SORTORDER_ASCENDING = _yappi.SORTORDER_ASCENDING
SORTORDER_DESCENDING = _yappi.SORTORDER_DESCENDING
SHOW_ALL = _yappi.SHOW_ALL
'''
__callback will only be called once per-thread. _yappi will detect
the new thread and changes the profilefunc param of the ThreadState
structure. This is an internal function please don't mess with it.
'''
def __callback(frame, event, arg):
_yappi.profile_event(frame, event, arg)
return __callback
'''
...
Args:
builtins: If set true, then builtin functions are profiled too.
timing_sample: will cause the profiler to do timing measuresements
according to the value. Will increase profiler speed but
decrease accuracy.
'''
def start(builtins = False, timing_sample=1):
threading.setprofile(__callback)
_yappi.start(builtins, timing_sample)
def stop():
threading.setprofile(None)
_yappi.stop()
def enum_stats(fenum):
_yappi.enum_stats(fenum)
def get_stats(sorttype=_yappi.SORTTYPE_NCALL,
sortorder=_yappi.SORTORDER_DESCENDING,
limit=_yappi.SHOW_ALL):
return _yappi.get_stats(sorttype, sortorder, limit)
def print_stats(sorttype=_yappi.SORTTYPE_NCALL,
sortorder=_yappi.SORTORDER_DESCENDING,
limit=_yappi.SHOW_ALL):
li = get_stats(sorttype, sortorder, limit)
for it in li:
print it
def clear_stats():
_yappi.clear_stats()
if __name__ != "__main__":
pass