Handle keyboard interrupts properly.

This commit is contained in:
InValidFire 2021-03-31 14:53:01 -04:00
parent 58ca231454
commit bb6b157774
1 changed files with 14 additions and 8 deletions

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,6 +22,7 @@ def start_bot():
bot = start_bot() bot = start_bot()
try:
while True: while True:
if bot.poll() is not None: if bot.poll() is not None:
if bot.returncode == 26: if bot.returncode == 26:
@ -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")