Handle SIGINT more gracefully

kittens/hyperlinked_grep/main.py

@@ -4,6 +4,7 @@
 
 import os
 import re
+import signal
 import socket
 import subprocess
 import sys
@@ -12,14 +13,11 @@ from urllib.parse import quote_from_bytes
 
 
 def write_hyperlink(write: Callable[[bytes], None], url: bytes, line: bytes, frag: bytes = b'') -> None:
-    write(b'\033]8;;')
-    write(url)
+    text = b'\033]8;;' + url
     if frag:
-        write(b'#')
-        write(frag)
-    write(b'\033\\')
-    write(line)
-    write(b'\033]8;;\033\\')
+        text += b'#' + frag
+    text += b'\033\\' + line + b'\033]8;;\033\\'
+    write(text)
 
 
 def main() -> None:
@@ -36,6 +34,7 @@ def main() -> None:
     in_result: bytes = b''
     hostname = socket.gethostname().encode('utf-8')
 
+    try:
         for line in p.stdout:
             line = osc_pat.sub(b'', line)  # remove any existing hyperlinks
             clean_line = sgr_pat.sub(b'', line).rstrip()  # remove SGR formatting
@@ -54,6 +53,9 @@ def main() -> None:
                     write_hyperlink(write, in_result, line)
                 else:
                     write(line)
+    except KeyboardInterrupt:
+        p.send_signal(signal.SIGINT)
+        p.stdout.close()
     raise SystemExit(p.wait())