diff --git a/main.py b/main.py index b7f5c8f..7f637e7 100644 --- a/main.py +++ b/main.py @@ -4,6 +4,7 @@ import subprocess import logging import sys import time +import psutil logger = logging.getLogger(__name__) logger.setLevel(logging.DEBUG) @@ -21,11 +22,16 @@ def start_bot(): bot = start_bot() -while True: - if bot.poll() is not None: - if bot.returncode == 26: - logger.info("exit code 26 received, restarting bot!") - bot = start_bot() - else: - break - time.sleep(1) # keeps code from overworking. +try: + while True: + if bot.poll() is not None: + if bot.returncode == 26: + logger.info("exit code 26 received, restarting bot!") + bot = start_bot() + else: + break + time.sleep(1) # keeps code from overworking. +except KeyboardInterrupt: + print("Killing Bot Process") + psutil.Process(bot.pid).kill() + print("Killed successfully")