VimPlug/jedi
1
0
Fork 1
mirror of https://github.com/davidhalter/jedi.git synced 2025-12-07 22:44:27 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Improve the profiling script

Browse Source
This commit is contained in:
Dave Halter
2018-11-28 09:52:31 +01:00
parent 5e6e4356fc
commit 2eb5e9b42d
1 changed files with 16 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
scripts/profile_output.py
View File

@@ -1,10 +1,10 @@
#!/usr/bin/env python #!/usr/bin/env python3.6
""" """
Profile a piece of Python code with ``cProfile``. Tries a completion on a Profile a piece of Python code with ``profile``. Tries a completion on a
certain piece of code. certain piece of code.
Usage: Usage:
profile.py [<code>] [-n <number>] [-d] [-o] [-s <sort>] profile.py [<code>] [-n <number>] [-d] [-o] [-s <sort>] [-i]
profile.py -h | --help profile.py -h | --help
Options: Options:
@@ -12,34 +12,41 @@ Options:
-n <number> Number of passes before profiling [default: 1]. -n <number> Number of passes before profiling [default: 1].
-d --debug Enable Jedi internal debugging. -d --debug Enable Jedi internal debugging.
-o --omit Omit profiler, just do a normal run. -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]. -s <sort> Sort the profile results, e.g. cum, name [default: time].
""" """
import time import time
import cProfile import profile
from docopt import docopt from docopt import docopt
import jedi import jedi
def run(code, index): def run(code, index, infer=False):
start = time.time() start = time.time()
result = jedi.Script(code).completions() 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)) print('Used %ss for the %sth run.' % (time.time() - start, index + 1))
return result return result
def main(args): def main(args):
code = args['<code>'] code = args['<code>']
infer = args['--infer']
n = int(args['-n']) n = int(args['-n'])
for i in range(n): for i in range(n):
run(code, i) run(code, i, infer=infer)
jedi.set_debug_function(notices=args['--debug']) jedi.set_debug_function(notices=args['--debug'])
if args['--omit']: if args['--omit']:
run(code, n) run(code, n, infer=infer)
else: else:
cProfile.runctx('run(code, n)', globals(), locals(), sort=args['-s']) profile.runctx('run(code, n, infer=infer)', globals(), locals(), sort=args['-s'])
if __name__ == '__main__': if __name__ == '__main__':