JSON renderer

pyinstrument/profiler.py

@@ -114,5 +114,7 @@ def get_renderer_class(name):
         return renderers.ConsoleRenderer
     elif name == 'html':
         return renderers.HTMLRenderer
+    elif name == 'json':
+        return renderers.JSONRenderer
     else:
         return object_with_import_path(name)

pyinstrument/renderers.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 import abc
 import os
+import json
 from . import six
 
 class Renderer(object):
@@ -128,6 +129,21 @@ def render_frame(self, frame):
         result += '</div></div>'
 
         return result
+
+def _json_frame(frame):
+    return {
+        'function': frame.function,
+        'file_path': frame.file_path_short,
+        'line_no': frame.line_no,
+        'time': frame.time(),
+        'children': [_json_frame(frame) for frame in frame.children]
+    }
+
+class JSONRenderer(Renderer):
+    preferred_recorder = 'time_aggregating'
+
+    def render(self, frame):
+        return json.dumps(_json_frame(frame))
 
 
 class colors_enabled: