| pulkit |
| hg-reviewers |
These fields are str on Python 2 and 3. This module doesn't import any Mercurial
modules. So I just did the str -> bytes inline.
| Automatic diff as part of commit; lint not applicable. |
| Automatic diff as part of commit; unit tests not applicable. |
| Path | Packages | |||
|---|---|---|---|---|
| M | mercurial/lsprof.py (10 lines) |
| se = e.calls[j] | se = e.calls[j] | ||||
| if not isinstance(se.code, str): | if not isinstance(se.code, str): | ||||
| e.calls[j] = type(se)((label(se.code),) + se[1:]) | e.calls[j] = type(se)((label(se.code),) + se[1:]) | ||||
| _fn2mod = {} | _fn2mod = {} | ||||
| def label(code): | def label(code): | ||||
| if isinstance(code, str): | if isinstance(code, str): | ||||
| if sys.version_info.major >= 3: | |||||
| code = code.encode('latin-1') | |||||
| return code | return code | ||||
| try: | try: | ||||
| mname = _fn2mod[code.co_filename] | mname = _fn2mod[code.co_filename] | ||||
| except KeyError: | except KeyError: | ||||
| for k, v in list(sys.modules.iteritems()): | for k, v in list(sys.modules.iteritems()): | ||||
| if v is None: | if v is None: | ||||
| continue | continue | ||||
| if not isinstance(getattr(v, '__file__', None), str): | if not isinstance(getattr(v, '__file__', None), str): | ||||
| continue | continue | ||||
| if v.__file__.startswith(code.co_filename): | if v.__file__.startswith(code.co_filename): | ||||
| mname = _fn2mod[code.co_filename] = k | mname = _fn2mod[code.co_filename] = k | ||||
| break | break | ||||
| else: | else: | ||||
| mname = _fn2mod[code.co_filename] = '<%s>' % code.co_filename | mname = _fn2mod[code.co_filename] = r'<%s>' % code.co_filename | ||||
| return '%s:%d(%s)' % (mname, code.co_firstlineno, code.co_name) | res = r'%s:%d(%s)' % (mname, code.co_firstlineno, code.co_name) | ||||
| if sys.version_info.major >= 3: | |||||
| res = res.encode('latin-1') | |||||
| return res | |||||
| if __name__ == '__main__': | if __name__ == '__main__': | ||||
| import os | import os | ||||
| sys.argv = sys.argv[1:] | sys.argv = sys.argv[1:] | ||||
| if not sys.argv: | if not sys.argv: | ||||
| print("usage: lsprof.py <script> <arguments...>", file=sys.stderr) | print("usage: lsprof.py <script> <arguments...>", file=sys.stderr) | ||||
| sys.exit(2) | sys.exit(2) | ||||
| sys.path.insert(0, os.path.abspath(os.path.dirname(sys.argv[0]))) | sys.path.insert(0, os.path.abspath(os.path.dirname(sys.argv[0]))) | ||||
| stats = profile(execfile, sys.argv[0], globals(), locals()) | stats = profile(execfile, sys.argv[0], globals(), locals()) | ||||
| stats.sort() | stats.sort() | ||||
| stats.pprint() | stats.pprint() | ||||