From 05535e04f1a397e3a6d585ecf24b28c595437ad4 Mon Sep 17 00:00:00 2001
From: Dave Halter <davidhalter88@gmail.com>
Date: Thu, 11 May 2017 08:31:19 -0400
Subject: [PATCH] Add a diff parser profile script from Jedi.

---
 scripts/diff_parser_profile.py | 50 ++++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)
 create mode 100755 scripts/diff_parser_profile.py

diff --git a/scripts/diff_parser_profile.py b/scripts/diff_parser_profile.py
new file mode 100755
index 0000000..a152a3e
--- /dev/null
+++ b/scripts/diff_parser_profile.py
@@ -0,0 +1,50 @@
+#!/usr/bin/env python
+"""
+Profile a piece of Python code with ``cProfile`` that uses the diff parser.
+
+Usage:
+  profile.py <file> [-d] [-s <sort>]
+  profile.py -h | --help
+
+Options:
+  -h --help     Show this screen.
+  -d --debug    Enable Jedi internal debugging.
+  -s <sort>     Sort the profile results, e.g. cumtime, name [default: time].
+"""
+
+import cProfile
+
+from docopt import docopt
+from jedi.parser.python import load_grammar
+from jedi.parser.diff import DiffParser
+from jedi.parser.python import ParserWithRecovery
+from jedi._compatibility import u
+from jedi.common import splitlines
+import jedi
+
+
+def run(parser, lines):
+    diff_parser = DiffParser(parser)
+    diff_parser.update(lines)
+    # Make sure used_names is loaded
+    parser.module.used_names
+
+
+def main(args):
+    if args['--debug']:
+        jedi.set_debug_function(notices=True)
+
+    with open(args['<file>']) as f:
+        code = f.read()
+    grammar = load_grammar()
+    parser = ParserWithRecovery(grammar, u(code))
+    # Make sure used_names is loaded
+    parser.module.used_names
+
+    code =  code + '\na\n'  # Add something so the diff parser needs to run.
+    lines = splitlines(code, keepends=True)
+    cProfile.runctx('run(parser, lines)', globals(), locals(), sort=args['-s'])
+
+if __name__ == '__main__':
+    args = docopt(__doc__)
+    main(args)