add simple interactive summary support

1 files changed, 7 insertions, 0 deletions

diff --git a/summarize.py b/summarize.py
index 46ca999..ca1927a 100755
--- a/summarize.py
+++ b/summarize.py
@@ -172,6 +172,9 @@ if __name__ == "__main__":
     parser.add_argument("--latex-preamble",
                         help="latex code to include in the preamble of generated standalones",
                         type=str)
+    parser.add_argument("-i", "--interactive",
+                        help="drop into repl after summarizing the results",
+                        action='store_true')
 
     args = parser.parse_args()
 
@@ -197,3 +200,7 @@ if __name__ == "__main__":
 
     summarize(benchmarks=args.benchmarks,
               exclude_benchmarks=args.exclude_benchmarks)
+
+    if args.interactive:
+        import code
+        code.InteractiveConsole(locals=globals()).interact()