Handle keyboard interrupts properly.

This commit is contained in:
InValidFire 2021-03-31 14:53:01 -04:00
parent 58ca231454
commit bb6b157774

View file

@ -4,6 +4,7 @@ import subprocess
import logging import logging
import sys import sys
import time import time
import psutil
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
logger.setLevel(logging.DEBUG) logger.setLevel(logging.DEBUG)
@ -21,7 +22,8 @@ def start_bot():
bot = start_bot() bot = start_bot()
while True: try:
while True:
if bot.poll() is not None: if bot.poll() is not None:
if bot.returncode == 26: if bot.returncode == 26:
logger.info("exit code 26 received, restarting bot!") logger.info("exit code 26 received, restarting bot!")
@ -29,3 +31,7 @@ while True:
else: else:
break break
time.sleep(1) # keeps code from overworking. time.sleep(1) # keeps code from overworking.
except KeyboardInterrupt:
print("Killing Bot Process")
psutil.Process(bot.pid).kill()
print("Killed successfully")