##// END OF EJS Templates
lsprof: use print function
Gregory Szorc -
r27617:b1a59b80 default
parent child Browse files
Show More
@@ -1,121 +1,121 b''
1 from __future__ import absolute_import
1 from __future__ import absolute_import, print_function
2 2
3 3 import _lsprof
4 4 import sys
5 5
6 6 Profiler = _lsprof.Profiler
7 7
8 8 # PyPy doesn't expose profiler_entry from the module.
9 9 profiler_entry = getattr(_lsprof, 'profiler_entry', None)
10 10
11 11 __all__ = ['profile', 'Stats']
12 12
13 13 def profile(f, *args, **kwds):
14 14 """XXX docstring"""
15 15 p = Profiler()
16 16 p.enable(subcalls=True, builtins=True)
17 17 try:
18 18 f(*args, **kwds)
19 19 finally:
20 20 p.disable()
21 21 return Stats(p.getstats())
22 22
23 23
24 24 class Stats(object):
25 25 """XXX docstring"""
26 26
27 27 def __init__(self, data):
28 28 self.data = data
29 29
30 30 def sort(self, crit="inlinetime"):
31 31 """XXX docstring"""
32 32 # profiler_entries isn't defined when running under PyPy.
33 33 if profiler_entry:
34 34 if crit not in profiler_entry.__dict__:
35 35 raise ValueError("Can't sort by %s" % crit)
36 36 elif self.data and not getattr(self.data[0], crit, None):
37 37 raise ValueError("Can't sort by %s" % crit)
38 38
39 39 self.data.sort(key=lambda x: getattr(x, crit), reverse=True)
40 40 for e in self.data:
41 41 if e.calls:
42 42 e.calls.sort(key=lambda x: getattr(x, crit), reverse=True)
43 43
44 44 def pprint(self, top=None, file=None, limit=None, climit=None):
45 45 """XXX docstring"""
46 46 if file is None:
47 47 file = sys.stdout
48 48 d = self.data
49 49 if top is not None:
50 50 d = d[:top]
51 51 cols = "% 12s %12s %11.4f %11.4f %s\n"
52 52 hcols = "% 12s %12s %12s %12s %s\n"
53 53 file.write(hcols % ("CallCount", "Recursive", "Total(s)",
54 54 "Inline(s)", "module:lineno(function)"))
55 55 count = 0
56 56 for e in d:
57 57 file.write(cols % (e.callcount, e.reccallcount, e.totaltime,
58 58 e.inlinetime, label(e.code)))
59 59 count += 1
60 60 if limit is not None and count == limit:
61 61 return
62 62 ccount = 0
63 63 if climit and e.calls:
64 64 for se in e.calls:
65 65 file.write(cols % (se.callcount, se.reccallcount,
66 66 se.totaltime, se.inlinetime,
67 67 " %s" % label(se.code)))
68 68 count += 1
69 69 ccount += 1
70 70 if limit is not None and count == limit:
71 71 return
72 72 if climit is not None and ccount == climit:
73 73 break
74 74
75 75 def freeze(self):
76 76 """Replace all references to code objects with string
77 77 descriptions; this makes it possible to pickle the instance."""
78 78
79 79 # this code is probably rather ickier than it needs to be!
80 80 for i in range(len(self.data)):
81 81 e = self.data[i]
82 82 if not isinstance(e.code, str):
83 83 self.data[i] = type(e)((label(e.code),) + e[1:])
84 84 if e.calls:
85 85 for j in range(len(e.calls)):
86 86 se = e.calls[j]
87 87 if not isinstance(se.code, str):
88 88 e.calls[j] = type(se)((label(se.code),) + se[1:])
89 89
90 90 _fn2mod = {}
91 91
92 92 def label(code):
93 93 if isinstance(code, str):
94 94 return code
95 95 try:
96 96 mname = _fn2mod[code.co_filename]
97 97 except KeyError:
98 98 for k, v in list(sys.modules.iteritems()):
99 99 if v is None:
100 100 continue
101 101 if not isinstance(getattr(v, '__file__', None), str):
102 102 continue
103 103 if v.__file__.startswith(code.co_filename):
104 104 mname = _fn2mod[code.co_filename] = k
105 105 break
106 106 else:
107 107 mname = _fn2mod[code.co_filename] = '<%s>' % code.co_filename
108 108
109 109 return '%s:%d(%s)' % (mname, code.co_firstlineno, code.co_name)
110 110
111 111
112 112 if __name__ == '__main__':
113 113 import os
114 114 sys.argv = sys.argv[1:]
115 115 if not sys.argv:
116 print >> sys.stderr, "usage: lsprof.py <script> <arguments...>"
116 print("usage: lsprof.py <script> <arguments...>", file=sys.stderr)
117 117 sys.exit(2)
118 118 sys.path.insert(0, os.path.abspath(os.path.dirname(sys.argv[0])))
119 119 stats = profile(execfile, sys.argv[0], globals(), locals())
120 120 stats.sort()
121 121 stats.pprint()
@@ -1,184 +1,183 b''
1 1 #require test-repo
2 2
3 3 $ cd "$TESTDIR"/..
4 4
5 5 $ hg files 'set:(**.py)' | sed 's|\\|/|g' | xargs python contrib/check-py3-compat.py
6 6 contrib/casesmash.py not using absolute_import
7 7 contrib/check-code.py not using absolute_import
8 8 contrib/check-code.py requires print_function
9 9 contrib/check-config.py not using absolute_import
10 10 contrib/check-config.py requires print_function
11 11 contrib/debugcmdserver.py not using absolute_import
12 12 contrib/debugcmdserver.py requires print_function
13 13 contrib/debugshell.py not using absolute_import
14 14 contrib/fixpax.py not using absolute_import
15 15 contrib/fixpax.py requires print_function
16 16 contrib/hgclient.py not using absolute_import
17 17 contrib/hgclient.py requires print_function
18 18 contrib/hgfixes/fix_bytes.py not using absolute_import
19 19 contrib/hgfixes/fix_bytesmod.py not using absolute_import
20 20 contrib/hgfixes/fix_leftover_imports.py not using absolute_import
21 21 contrib/import-checker.py not using absolute_import
22 22 contrib/import-checker.py requires print_function
23 23 contrib/memory.py not using absolute_import
24 24 contrib/perf.py not using absolute_import
25 25 contrib/python-hook-examples.py not using absolute_import
26 26 contrib/revsetbenchmarks.py not using absolute_import
27 27 contrib/revsetbenchmarks.py requires print_function
28 28 contrib/showstack.py not using absolute_import
29 29 contrib/synthrepo.py not using absolute_import
30 30 contrib/win32/hgwebdir_wsgi.py not using absolute_import
31 31 doc/check-seclevel.py not using absolute_import
32 32 doc/gendoc.py not using absolute_import
33 33 doc/hgmanpage.py not using absolute_import
34 34 hgext/__init__.py not using absolute_import
35 35 hgext/acl.py not using absolute_import
36 36 hgext/blackbox.py not using absolute_import
37 37 hgext/bugzilla.py not using absolute_import
38 38 hgext/censor.py not using absolute_import
39 39 hgext/children.py not using absolute_import
40 40 hgext/churn.py not using absolute_import
41 41 hgext/clonebundles.py not using absolute_import
42 42 hgext/color.py not using absolute_import
43 43 hgext/convert/__init__.py not using absolute_import
44 44 hgext/convert/bzr.py not using absolute_import
45 45 hgext/convert/common.py not using absolute_import
46 46 hgext/convert/convcmd.py not using absolute_import
47 47 hgext/convert/cvs.py not using absolute_import
48 48 hgext/convert/cvsps.py not using absolute_import
49 49 hgext/convert/darcs.py not using absolute_import
50 50 hgext/convert/filemap.py not using absolute_import
51 51 hgext/convert/git.py not using absolute_import
52 52 hgext/convert/gnuarch.py not using absolute_import
53 53 hgext/convert/hg.py not using absolute_import
54 54 hgext/convert/monotone.py not using absolute_import
55 55 hgext/convert/p4.py not using absolute_import
56 56 hgext/convert/subversion.py not using absolute_import
57 57 hgext/convert/transport.py not using absolute_import
58 58 hgext/eol.py not using absolute_import
59 59 hgext/extdiff.py not using absolute_import
60 60 hgext/factotum.py not using absolute_import
61 61 hgext/fetch.py not using absolute_import
62 62 hgext/gpg.py not using absolute_import
63 63 hgext/graphlog.py not using absolute_import
64 64 hgext/hgcia.py not using absolute_import
65 65 hgext/hgk.py not using absolute_import
66 66 hgext/highlight/__init__.py not using absolute_import
67 67 hgext/highlight/highlight.py not using absolute_import
68 68 hgext/histedit.py not using absolute_import
69 69 hgext/keyword.py not using absolute_import
70 70 hgext/largefiles/__init__.py not using absolute_import
71 71 hgext/largefiles/basestore.py not using absolute_import
72 72 hgext/largefiles/lfcommands.py not using absolute_import
73 73 hgext/largefiles/lfutil.py not using absolute_import
74 74 hgext/largefiles/localstore.py not using absolute_import
75 75 hgext/largefiles/overrides.py not using absolute_import
76 76 hgext/largefiles/proto.py not using absolute_import
77 77 hgext/largefiles/remotestore.py not using absolute_import
78 78 hgext/largefiles/reposetup.py not using absolute_import
79 79 hgext/largefiles/uisetup.py not using absolute_import
80 80 hgext/largefiles/wirestore.py not using absolute_import
81 81 hgext/mq.py not using absolute_import
82 82 hgext/notify.py not using absolute_import
83 83 hgext/pager.py not using absolute_import
84 84 hgext/patchbomb.py not using absolute_import
85 85 hgext/purge.py not using absolute_import
86 86 hgext/rebase.py not using absolute_import
87 87 hgext/record.py not using absolute_import
88 88 hgext/relink.py not using absolute_import
89 89 hgext/schemes.py not using absolute_import
90 90 hgext/share.py not using absolute_import
91 91 hgext/shelve.py not using absolute_import
92 92 hgext/strip.py not using absolute_import
93 93 hgext/transplant.py not using absolute_import
94 94 hgext/win32mbcs.py not using absolute_import
95 95 hgext/win32text.py not using absolute_import
96 96 hgext/zeroconf/Zeroconf.py not using absolute_import
97 97 hgext/zeroconf/Zeroconf.py requires print_function
98 98 hgext/zeroconf/__init__.py not using absolute_import
99 99 i18n/check-translation.py not using absolute_import
100 100 i18n/polib.py not using absolute_import
101 101 mercurial/cmdutil.py not using absolute_import
102 102 mercurial/commands.py not using absolute_import
103 mercurial/lsprof.py requires print_function
104 103 mercurial/lsprofcalltree.py requires print_function
105 104 mercurial/mail.py requires print_function
106 105 setup.py not using absolute_import
107 106 tests/filterpyflakes.py requires print_function
108 107 tests/generate-working-copy-states.py requires print_function
109 108 tests/get-with-headers.py requires print_function
110 109 tests/heredoctest.py requires print_function
111 110 tests/hypothesishelpers.py not using absolute_import
112 111 tests/hypothesishelpers.py requires print_function
113 112 tests/killdaemons.py not using absolute_import
114 113 tests/md5sum.py not using absolute_import
115 114 tests/mockblackbox.py not using absolute_import
116 115 tests/printenv.py not using absolute_import
117 116 tests/readlink.py not using absolute_import
118 117 tests/readlink.py requires print_function
119 118 tests/revlog-formatv0.py not using absolute_import
120 119 tests/run-tests.py not using absolute_import
121 120 tests/seq.py not using absolute_import
122 121 tests/seq.py requires print_function
123 122 tests/silenttestrunner.py not using absolute_import
124 123 tests/silenttestrunner.py requires print_function
125 124 tests/sitecustomize.py not using absolute_import
126 125 tests/svn-safe-append.py not using absolute_import
127 126 tests/svnxml.py not using absolute_import
128 127 tests/test-ancestor.py requires print_function
129 128 tests/test-atomictempfile.py not using absolute_import
130 129 tests/test-batching.py not using absolute_import
131 130 tests/test-batching.py requires print_function
132 131 tests/test-bdiff.py not using absolute_import
133 132 tests/test-bdiff.py requires print_function
134 133 tests/test-context.py not using absolute_import
135 134 tests/test-context.py requires print_function
136 135 tests/test-demandimport.py not using absolute_import
137 136 tests/test-demandimport.py requires print_function
138 137 tests/test-dispatch.py not using absolute_import
139 138 tests/test-dispatch.py requires print_function
140 139 tests/test-doctest.py not using absolute_import
141 140 tests/test-duplicateoptions.py not using absolute_import
142 141 tests/test-duplicateoptions.py requires print_function
143 142 tests/test-filecache.py not using absolute_import
144 143 tests/test-filecache.py requires print_function
145 144 tests/test-filelog.py not using absolute_import
146 145 tests/test-filelog.py requires print_function
147 146 tests/test-hg-parseurl.py not using absolute_import
148 147 tests/test-hg-parseurl.py requires print_function
149 148 tests/test-hgweb-auth.py not using absolute_import
150 149 tests/test-hgweb-auth.py requires print_function
151 150 tests/test-hgwebdir-paths.py not using absolute_import
152 151 tests/test-hybridencode.py not using absolute_import
153 152 tests/test-hybridencode.py requires print_function
154 153 tests/test-lrucachedict.py not using absolute_import
155 154 tests/test-lrucachedict.py requires print_function
156 155 tests/test-manifest.py not using absolute_import
157 156 tests/test-minirst.py not using absolute_import
158 157 tests/test-minirst.py requires print_function
159 158 tests/test-parseindex2.py not using absolute_import
160 159 tests/test-parseindex2.py requires print_function
161 160 tests/test-pathencode.py not using absolute_import
162 161 tests/test-pathencode.py requires print_function
163 162 tests/test-propertycache.py not using absolute_import
164 163 tests/test-propertycache.py requires print_function
165 164 tests/test-revlog-ancestry.py not using absolute_import
166 165 tests/test-revlog-ancestry.py requires print_function
167 166 tests/test-run-tests.py not using absolute_import
168 167 tests/test-simplemerge.py not using absolute_import
169 168 tests/test-status-inprocess.py not using absolute_import
170 169 tests/test-status-inprocess.py requires print_function
171 170 tests/test-symlink-os-yes-fs-no.py not using absolute_import
172 171 tests/test-trusted.py not using absolute_import
173 172 tests/test-trusted.py requires print_function
174 173 tests/test-ui-color.py not using absolute_import
175 174 tests/test-ui-color.py requires print_function
176 175 tests/test-ui-config.py not using absolute_import
177 176 tests/test-ui-config.py requires print_function
178 177 tests/test-ui-verbosity.py not using absolute_import
179 178 tests/test-ui-verbosity.py requires print_function
180 179 tests/test-url.py not using absolute_import
181 180 tests/test-url.py requires print_function
182 181 tests/test-walkrepo.py requires print_function
183 182 tests/test-wireproto.py requires print_function
184 183 tests/tinyproxy.py requires print_function
General Comments 0
You need to be logged in to leave comments. Login now