Move disabling signals to own method

1 files changed, 9 insertions, 7 deletions

diff --git a/mini-scalene.py b/mini-scalene.py
index 4d3909b..5d8f6b2 100644
--- a/mini-scalene.py
+++ b/mini-scalene.py
@@ -58,9 +58,7 @@ class mini_scalene:
     @staticmethod
     def exit_handler():
         '''Turn off our profiling signals and pretty-print profiling information.'''
-        signal.signal(signal.ITIMER_PROF, signal.SIG_IGN)
-        signal.signal(signal.SIGVTALRM, signal.SIG_IGN)
-        signal.setitimer(signal.ITIMER_PROF, 0)
+        mini_scalene.disable_signals()
         # If we've collected any samples, dump them.
         print("CPU usage:")
         if mini_scalene.total_cpu_samples > 0:
@@ -79,6 +77,12 @@ class mini_scalene:
             print("(did not run long enough to profile)")
 
     @staticmethod
+    def disable_signals():
+        signal.signal(signal.ITIMER_PROF, signal.SIG_IGN)
+        signal.signal(signal.SIGVTALRM, signal.SIG_IGN)
+        signal.setitimer(signal.ITIMER_PROF, 0)
+
+    @staticmethod
     def cpu_signal_handler(sig, frame):
         keys = mini_scalene.compute_frames_to_record(frame)
         for key in keys:
@@ -145,11 +149,9 @@ class mini_scalene:
         # FIXME Assume GuixSD. Makes filtering easy
         if '/gnu/store' in filename:
             return False
-        if 'mini-scalene.py' in filename:
+        if filename[0] == '<':
             return False
-        if '<frozen importlib._bootstrap>' in filename:
-            return False
-        if '<frozen importlib._bootstrap_external>' in filename:
+        if 'mini-scalene.py' in filename:
             return False
         return True