mirror of
https://github.com/davidhalter/jedi.git
synced 2025-12-07 06:24:27 +08:00
57 lines
1.4 KiB
Python
Executable File
57 lines
1.4 KiB
Python
Executable File
#!/usr/bin/env python3.6
|
|
"""
|
|
Profile a piece of Python code with ``profile``. Tries a completion on a
|
|
certain piece of code.
|
|
|
|
Usage:
|
|
profile.py [<code>] [-n <number>] [-d] [-o] [-s <sort>] [-i]
|
|
profile.py -h | --help
|
|
|
|
Options:
|
|
-h --help Show this screen.
|
|
-n <number> Number of passes before profiling [default: 1].
|
|
-d --debug Enable Jedi internal debugging.
|
|
-o --omit Omit profiler, just do a normal run.
|
|
-i --infer Infer types instead of completions.
|
|
-s <sort> Sort the profile results, e.g. cum, name [default: time].
|
|
"""
|
|
|
|
import time
|
|
import profile
|
|
|
|
from docopt import docopt
|
|
import jedi
|
|
|
|
|
|
def run(code, index, infer=False):
|
|
start = time.time()
|
|
script = jedi.Script(code)
|
|
if infer:
|
|
result = script.goto_definitions()
|
|
else:
|
|
result = script.completions()
|
|
print('Used %ss for the %sth run.' % (time.time() - start, index + 1))
|
|
return result
|
|
|
|
|
|
def main(args):
|
|
code = args['<code>']
|
|
infer = args['--infer']
|
|
n = int(args['-n'])
|
|
|
|
for i in range(n):
|
|
run(code, i, infer=infer)
|
|
|
|
jedi.set_debug_function(notices=args['--debug'])
|
|
if args['--omit']:
|
|
run(code, n, infer=infer)
|
|
else:
|
|
profile.runctx('run(code, n, infer=infer)', globals(), locals(), sort=args['-s'])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
args = docopt(__doc__)
|
|
if args['<code>'] is None:
|
|
args['<code>'] = 'import numpy; numpy.array([0]).'
|
|
main(args)
|