From f7c41ef8bb7993d09aa54cd9b3773a55a662ffb2 Mon Sep 17 00:00:00 2001 From: bd Date: Tue, 10 Jun 2025 16:05:01 -0400 Subject: Move disabling signals to own method --- mini-scalene.py | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) (limited to 'mini-scalene.py') 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: @@ -78,6 +76,12 @@ class mini_scalene: else: 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) @@ -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 '' in filename: - return False - if '' in filename: + if 'mini-scalene.py' in filename: return False return True -- cgit v1.2.3