Added SIGINT handler.

get-prebuilt.py

@@ -4,10 +4,12 @@
 import sys
 import json
 import urllib.request
-from utils import Fore, parse_image_arg, chunked_copy, clear_progress
+from utils import Fore, parse_image_arg, chunked_copy, clear_progress, handle_sigint
 
 # handle arguments
 
+handle_sigint()
+
 if len(sys.argv) < 2:
 	print('usage: ./get-prebuilt.py image[:tag]')
 	sys.exit(-1)

get-source.py

@@ -2,10 +2,12 @@
 # coding=utf-8
 import sys
 import urllib.request
-from utils import Fore, parse_image_arg, chunked_copy, clear_progress
+from utils import Fore, parse_image_arg, chunked_copy, clear_progress, handle_sigint
 
 # handle arguments
 
+handle_sigint()
+
 if len(sys.argv) < 2:
 	print('usage: ./get-source.py image[:tag]')
 	sys.exit(-1)

install.py

@@ -22,6 +22,8 @@
 
 # handle arguments
 
+handle_sigint()
+
 imgarg   = ''
 runhooks = True
 

switch.py

@@ -4,10 +4,12 @@
 import sys
 import os.path
 import subprocess
-from utils import Fore, parse_image_arg, probe_wsl, get_label, path_trans
+from utils import Fore, parse_image_arg, probe_wsl, get_label, path_trans, handle_sigint
 
 # handle arguments
 
+handle_sigint()
+
 if len(sys.argv) < 2:
 
 	# print usage information

utils.py

@@ -7,6 +7,7 @@
 import glob
 import time
 import shlex
+import signal
 import subprocess
 
 
@@ -55,6 +56,18 @@ class Fore:
 		RESET  = ''
 
 
+# registers for the interrupt signal in order to gracefully exit when Ctrl-C is hit
+
+def handle_sigint():
+	def signal_handler(signal, frame):
+		clear_progress()
+		show_cursor()
+		print('%s[!]%s Terminating early due to interruption.' % (Fore.RED, Fore.RESET))
+		sys.exit(-1)
+
+	signal.signal(signal.SIGINT, signal_handler)
+
+
 # parse image[:tag] | archive argument
 
 def parse_image_arg(argv, can_be_file = False):