Adding an ExitSignalHandler class.

1 files changed, 22 insertions, 0 deletions

diff --git a/src/argaze/utils/MiscFeatures.py b/src/argaze/utils/MiscFeatures.py
index 9c62d89..56a77bb 100644
--- a/src/argaze/utils/MiscFeatures.py
+++ b/src/argaze/utils/MiscFeatures.py
@@ -21,3 +21,25 @@ def printProgressBar (iteration:int, total:int, prefix:str = '', suffix:str = ''
     # Print New Line on Complete
     if iteration == total: 
         print()
+
+# Handle exit event
+class ExitSignalHandler():
+
+    def __init__(self):
+
+        import signal
+        import threading
+
+        global __exit_event
+        global __on_exit_signal
+
+        __exit_event = threading.Event()
+
+        def __on_exit_signal(signo, _frame):
+            __exit_event.set()
+
+        for sig in ('TERM', 'HUP', 'INT'):
+            signal.signal(getattr(signal, 'SIG'+sig), __on_exit_signal)
+
+    def status(self):
+        return __exit_event.is_set()
\ No newline at end of file