From c37a0c280049a3173a55a26813cb1157990dca00 Mon Sep 17 00:00:00 2001 From: Alejandro Aristizabal Date: Wed, 6 Dec 2023 11:21:23 -0500 Subject: [PATCH] Handle observer interruption --- cli/medperf/commands/dataset/create.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/cli/medperf/commands/dataset/create.py b/cli/medperf/commands/dataset/create.py index 20439fe4c..1adce1876 100644 --- a/cli/medperf/commands/dataset/create.py +++ b/cli/medperf/commands/dataset/create.py @@ -3,7 +3,6 @@ import sys import signal from pathlib import Path -from copy import deepcopy from medperf.entities.dataset import Dataset from medperf.enums import Status import medperf.config as config @@ -23,8 +22,9 @@ class ReportHandler(FileSystemEventHandler): - def __init__(self, preparation_obj: "DataPreparation"): + def __init__(self, preparation_obj: "DataPreparation", submission_approved): self.preparation = preparation_obj + self.submission_approved = submission_approved def on_created(self, event): self.on_modified(event) @@ -230,8 +230,8 @@ def sigint_handler(sig, frame): if approved: signal.signal(signal.SIGINT, sigint_handler) - observer.schedule(ReportHandler(self), self.out_path) - observer.start() + observer.schedule(ReportHandler(self, approved), self.out_path) + observer.start() self.ui.text = "Running preparation step..." try: @@ -261,7 +261,7 @@ def sigint_handler(sig, frame): self.ui.print("> Cube execution complete") # If any observer or signal was set, stop them - signal.pause() + signal.signal(signal.SIGINT, signal.default_int_handler) observer.stop() # Send a last update to indicate preparation process finished