Show More
@@ -1,121 +1,121 b'' | |||||
1 | from __future__ import absolute_import, print_function |
|
1 | from __future__ import absolute_import, print_function | |
2 |
|
2 | |||
3 | import _lsprof |
|
3 | import _lsprof | |
4 | import sys |
|
4 | import sys | |
5 |
|
5 | |||
6 | Profiler = _lsprof.Profiler |
|
6 | Profiler = _lsprof.Profiler | |
7 |
|
7 | |||
8 | # PyPy doesn't expose profiler_entry from the module. |
|
8 | # PyPy doesn't expose profiler_entry from the module. | |
9 | profiler_entry = getattr(_lsprof, 'profiler_entry', None) |
|
9 | profiler_entry = getattr(_lsprof, 'profiler_entry', None) | |
10 |
|
10 | |||
11 | __all__ = ['profile', 'Stats'] |
|
11 | __all__ = ['profile', 'Stats'] | |
12 |
|
12 | |||
13 | def profile(f, *args, **kwds): |
|
13 | def profile(f, *args, **kwds): | |
14 | """XXX docstring""" |
|
14 | """XXX docstring""" | |
15 | p = Profiler() |
|
15 | p = Profiler() | |
16 | p.enable(subcalls=True, builtins=True) |
|
16 | p.enable(subcalls=True, builtins=True) | |
17 | try: |
|
17 | try: | |
18 | f(*args, **kwds) |
|
18 | f(*args, **kwds) | |
19 | finally: |
|
19 | finally: | |
20 | p.disable() |
|
20 | p.disable() | |
21 | return Stats(p.getstats()) |
|
21 | return Stats(p.getstats()) | |
22 |
|
22 | |||
23 |
|
23 | |||
24 | class Stats(object): |
|
24 | class Stats(object): | |
25 | """XXX docstring""" |
|
25 | """XXX docstring""" | |
26 |
|
26 | |||
27 | def __init__(self, data): |
|
27 | def __init__(self, data): | |
28 | self.data = data |
|
28 | self.data = data | |
29 |
|
29 | |||
30 | def sort(self, crit=r"inlinetime"): |
|
30 | def sort(self, crit=r"inlinetime"): | |
31 | """XXX docstring""" |
|
31 | """XXX docstring""" | |
32 | # profiler_entries isn't defined when running under PyPy. |
|
32 | # profiler_entries isn't defined when running under PyPy. | |
33 | if profiler_entry: |
|
33 | if profiler_entry: | |
34 | if crit not in profiler_entry.__dict__: |
|
34 | if crit not in profiler_entry.__dict__: | |
35 | raise ValueError("Can't sort by %s" % crit) |
|
35 | raise ValueError("Can't sort by %s" % crit) | |
36 | elif self.data and not getattr(self.data[0], crit, None): |
|
36 | elif self.data and not getattr(self.data[0], crit, None): | |
37 | raise ValueError("Can't sort by %s" % crit) |
|
37 | raise ValueError("Can't sort by %s" % crit) | |
38 |
|
38 | |||
39 | self.data.sort(key=lambda x: getattr(x, crit), reverse=True) |
|
39 | self.data.sort(key=lambda x: getattr(x, crit), reverse=True) | |
40 | for e in self.data: |
|
40 | for e in self.data: | |
41 | if e.calls: |
|
41 | if e.calls: | |
42 | e.calls.sort(key=lambda x: getattr(x, crit), reverse=True) |
|
42 | e.calls.sort(key=lambda x: getattr(x, crit), reverse=True) | |
43 |
|
43 | |||
44 | def pprint(self, top=None, file=None, limit=None, climit=None): |
|
44 | def pprint(self, top=None, file=None, limit=None, climit=None): | |
45 | """XXX docstring""" |
|
45 | """XXX docstring""" | |
46 | if file is None: |
|
46 | if file is None: | |
47 | file = sys.stdout |
|
47 | file = sys.stdout | |
48 | d = self.data |
|
48 | d = self.data | |
49 | if top is not None: |
|
49 | if top is not None: | |
50 | d = d[:top] |
|
50 | d = d[:top] | |
51 |
cols = "% 12 |
|
51 | cols = "% 12d %12d %11.4f %11.4f %s\n" | |
52 | hcols = "% 12s %12s %12s %12s %s\n" |
|
52 | hcols = "% 12s %12s %12s %12s %s\n" | |
53 | file.write(hcols % ("CallCount", "Recursive", "Total(s)", |
|
53 | file.write(hcols % ("CallCount", "Recursive", "Total(s)", | |
54 | "Inline(s)", "module:lineno(function)")) |
|
54 | "Inline(s)", "module:lineno(function)")) | |
55 | count = 0 |
|
55 | count = 0 | |
56 | for e in d: |
|
56 | for e in d: | |
57 | file.write(cols % (e.callcount, e.reccallcount, e.totaltime, |
|
57 | file.write(cols % (e.callcount, e.reccallcount, e.totaltime, | |
58 | e.inlinetime, label(e.code))) |
|
58 | e.inlinetime, label(e.code))) | |
59 | count += 1 |
|
59 | count += 1 | |
60 | if limit is not None and count == limit: |
|
60 | if limit is not None and count == limit: | |
61 | return |
|
61 | return | |
62 | ccount = 0 |
|
62 | ccount = 0 | |
63 | if climit and e.calls: |
|
63 | if climit and e.calls: | |
64 | for se in e.calls: |
|
64 | for se in e.calls: | |
65 | file.write(cols % (se.callcount, se.reccallcount, |
|
65 | file.write(cols % (se.callcount, se.reccallcount, | |
66 | se.totaltime, se.inlinetime, |
|
66 | se.totaltime, se.inlinetime, | |
67 | " %s" % label(se.code))) |
|
67 | " %s" % label(se.code))) | |
68 | count += 1 |
|
68 | count += 1 | |
69 | ccount += 1 |
|
69 | ccount += 1 | |
70 | if limit is not None and count == limit: |
|
70 | if limit is not None and count == limit: | |
71 | return |
|
71 | return | |
72 | if climit is not None and ccount == climit: |
|
72 | if climit is not None and ccount == climit: | |
73 | break |
|
73 | break | |
74 |
|
74 | |||
75 | def freeze(self): |
|
75 | def freeze(self): | |
76 | """Replace all references to code objects with string |
|
76 | """Replace all references to code objects with string | |
77 | descriptions; this makes it possible to pickle the instance.""" |
|
77 | descriptions; this makes it possible to pickle the instance.""" | |
78 |
|
78 | |||
79 | # this code is probably rather ickier than it needs to be! |
|
79 | # this code is probably rather ickier than it needs to be! | |
80 | for i in range(len(self.data)): |
|
80 | for i in range(len(self.data)): | |
81 | e = self.data[i] |
|
81 | e = self.data[i] | |
82 | if not isinstance(e.code, str): |
|
82 | if not isinstance(e.code, str): | |
83 | self.data[i] = type(e)((label(e.code),) + e[1:]) |
|
83 | self.data[i] = type(e)((label(e.code),) + e[1:]) | |
84 | if e.calls: |
|
84 | if e.calls: | |
85 | for j in range(len(e.calls)): |
|
85 | for j in range(len(e.calls)): | |
86 | se = e.calls[j] |
|
86 | se = e.calls[j] | |
87 | if not isinstance(se.code, str): |
|
87 | if not isinstance(se.code, str): | |
88 | e.calls[j] = type(se)((label(se.code),) + se[1:]) |
|
88 | e.calls[j] = type(se)((label(se.code),) + se[1:]) | |
89 |
|
89 | |||
90 | _fn2mod = {} |
|
90 | _fn2mod = {} | |
91 |
|
91 | |||
92 | def label(code): |
|
92 | def label(code): | |
93 | if isinstance(code, str): |
|
93 | if isinstance(code, str): | |
94 | return code |
|
94 | return code | |
95 | try: |
|
95 | try: | |
96 | mname = _fn2mod[code.co_filename] |
|
96 | mname = _fn2mod[code.co_filename] | |
97 | except KeyError: |
|
97 | except KeyError: | |
98 | for k, v in list(sys.modules.iteritems()): |
|
98 | for k, v in list(sys.modules.iteritems()): | |
99 | if v is None: |
|
99 | if v is None: | |
100 | continue |
|
100 | continue | |
101 | if not isinstance(getattr(v, '__file__', None), str): |
|
101 | if not isinstance(getattr(v, '__file__', None), str): | |
102 | continue |
|
102 | continue | |
103 | if v.__file__.startswith(code.co_filename): |
|
103 | if v.__file__.startswith(code.co_filename): | |
104 | mname = _fn2mod[code.co_filename] = k |
|
104 | mname = _fn2mod[code.co_filename] = k | |
105 | break |
|
105 | break | |
106 | else: |
|
106 | else: | |
107 | mname = _fn2mod[code.co_filename] = '<%s>' % code.co_filename |
|
107 | mname = _fn2mod[code.co_filename] = '<%s>' % code.co_filename | |
108 |
|
108 | |||
109 | return '%s:%d(%s)' % (mname, code.co_firstlineno, code.co_name) |
|
109 | return '%s:%d(%s)' % (mname, code.co_firstlineno, code.co_name) | |
110 |
|
110 | |||
111 |
|
111 | |||
112 | if __name__ == '__main__': |
|
112 | if __name__ == '__main__': | |
113 | import os |
|
113 | import os | |
114 | sys.argv = sys.argv[1:] |
|
114 | sys.argv = sys.argv[1:] | |
115 | if not sys.argv: |
|
115 | if not sys.argv: | |
116 | print("usage: lsprof.py <script> <arguments...>", file=sys.stderr) |
|
116 | print("usage: lsprof.py <script> <arguments...>", file=sys.stderr) | |
117 | sys.exit(2) |
|
117 | sys.exit(2) | |
118 | sys.path.insert(0, os.path.abspath(os.path.dirname(sys.argv[0]))) |
|
118 | sys.path.insert(0, os.path.abspath(os.path.dirname(sys.argv[0]))) | |
119 | stats = profile(execfile, sys.argv[0], globals(), locals()) |
|
119 | stats = profile(execfile, sys.argv[0], globals(), locals()) | |
120 | stats.sort() |
|
120 | stats.sort() | |
121 | stats.pprint() |
|
121 | stats.pprint() |
General Comments 0
You need to be logged in to leave comments.
Login now