Exit program more gracefully on keyboard interrupt

clpo13 · Nov 17, 2023 · 0c93e4a · 0c93e4a
1 parent 2965c31
commit 0c93e4a
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/wikiget/wikiget.py b/src/wikiget/wikiget.py
@@ -138,5 +138,10 @@ def cli() -> None:
     logger.info("Starting download session using wikiget %s", wikiget.__version__)
     logger.debug("User agent: %s", wikiget.USER_AGENT)
 
-    exit_code = process_download(args)
-    sys.exit(exit_code)
+    try:
+        exit_code = process_download(args)
+    except KeyboardInterrupt:
+        logger.critical("Interrupted by user")
+        exit_code = 130
+    finally:
+        sys.exit(exit_code)