fix: make it easier to abort ongoing operation

vzhd1701 · May 7, 2022 · 830dc92 · 830dc92
1 parent 72a6a97
commit 830dc92
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/csv2notion/cli.py b/csv2notion/cli.py
@@ -1,5 +1,7 @@
 import argparse
 import logging
+import os
+import signal
 import sys
 from functools import partial
 from pathlib import Path
@@ -312,7 +314,14 @@ def critical_error(msg: str) -> None:
     sys.exit(1)
 
 
+def abort(*args):  # pragma: no cover
+    print("\nAbort")
+    os._exit(1)
+
+
 def main() -> None:
+    signal.signal(signal.SIGINT, abort)
+
     try:
         cli(sys.argv[1:])
     except (NotionError, CriticalError) as e: