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