diff --git a/nafas/__main__.py b/nafas/__main__.py index 4020589..d9bf68d 100644 --- a/nafas/__main__.py +++ b/nafas/__main__.py @@ -4,7 +4,7 @@ import argparse from nafas.functions import description_print, get_input_standard, input_filter from nafas.functions import get_program_data, program_description_print, run -from nafas.params import NAFAS_VERSION +from nafas.params import NAFAS_VERSION, EXIT_MESSAGE from art import tprint @@ -34,7 +34,11 @@ def main(): input("Press [R] to restart or any other key to exit.")) if INPUTINDEX.upper() != "R": EXIT_FLAG = True + print(EXIT_MESSAGE) if __name__ == "__main__": - main() + try: + main() + except KeyboardInterrupt: + print(EXIT_MESSAGE)