handling KeyboardInterrupt

AstroHuntsman · Aug 6, 2023 · 0d75e3c · 0d75e3c
1 parent 48c9544
commit 0d75e3c
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/scripts/sync_data_with_remote_server.py b/scripts/sync_data_with_remote_server.py
@@ -150,7 +150,16 @@ def transfer_data(self, filename, destination):
     )
     archiver.start()
 
-    while True:
-        if not archiver.is_running:
-            raise error.PanError("Archiver is no longer running.")
-        time.sleep(10)
+    try:
+        while True:
+            if not archiver.is_running:
+                raise error.PanError("Archiver is no longer running.")
+            time.sleep(10)
+    except (KeyboardInterrupt, SystemExit):
+        # Gracefully handle KeyboardInterrupt (Ctrl+C) and SystemExit during file transfer
+        archiver.logger.info("KeyboardInterrupt received. Stopping the RemoteArchiver gracefully...")
+        archiver.stop()
+
+        archiver.logger.info("Archiver stopped.")
+        # Make sure to exit the script properly after handling the KeyboardInterrupt or SystemExit
+        sys.exit(0)