Python: CG trace: Proper exception handling

RasmusWL · RasmusWL · commit d46b41011117 · 2020-07-20T01:22:33.000+02:00
diff --git a/python/tools/recorded-call-graph-metrics/src/cg_trace/tracer.py b/python/tools/recorded-call-graph-metrics/src/cg_trace/tracer.py
@@ -164,9 +164,10 @@ def run(self, code, globals, locals):
         try:
             sys.setprofile(self.profilefunc)
             exec(code, globals, locals)
-        # TODO: exception handling?
-        finally:
             sys.setprofile(None)
+        except Exception:
+            sys.setprofile(None)
+            LOGGER.info("Exception occurred while running program:", exc_info=True)
 
     def profilefunc(self, frame: FrameType, event: str, arg):
         # ignore everything until the first call, since that is `exec` from the `run`
