diff --git a/shellerator.py b/shellerator.py index 08cb38c..0c22069 100755 --- a/shellerator.py +++ b/shellerator.py @@ -21,17 +21,20 @@ import psutil from colorama import Fore from colorama import Style +from signal import signal +from signal import SIGINT import platform if platform.system() == 'Windows': from consolemenu import * else: from simple_term_menu import TerminalMenu -def signal_handler(sig, frame): +def ctrl_c_handler(sig, frame): + print(Fore.RED + Style.BRIGHT + "\n[!] Execution canceled by user." + Style.RESET_ALL + '\n') exit(1) # Handle Ctrl+C key interruption -signal.signal(signal.SIGINT, signal_handler) +signal(SIGINT, ctrl_c_handler) def menu(title, menu_list): if platform.system() == 'Windows':