diff --git a/octosuite b/octosuite index 73f0c9b..b5bfd97 100644 --- a/octosuite +++ b/octosuite @@ -2,19 +2,16 @@ import logging from src.main import * -from lib.colors import red,white,green,reset +from lib.colors import red,red_bg,white,white_bg,green_bg,green,reset if __name__ == '__main__': - while True: - try: - octosuite().on_start() - - except KeyboardInterrupt: - logging.info('Process interrupted with Ctrl+C') - exit(f'\n{white}[{red}x{white}] Process interrupted with {red}Ctrl{white}+{red}C{reset}') - - - except Exception as e: - logging.error(f'An error occured: {e}') - print(f'{white}[{red}!{white}] An error occured: {red}{e}{reset}') - input(f'\n{white}^ Press any key to continue{reset} ') + try: + octosuite().on_start() + + except KeyboardInterrupt: + logging.warning('Process interrupted with Ctrl+C') + exit(f'\n{white}[{red} x {white}] Process interrupted with {red_bg}Ctrl+C{reset}') + + except Exception as e: + logging.error(f'Session terminated on error: {e}') + exit(f'\n{white}[{red} ! {white}] Session {red_bg}terminated{reset}{white} on error: {red}{e}{reset}')