121 lines
3 KiB
Python
121 lines
3 KiB
Python
import datetime
|
|
import logging
|
|
import os
|
|
import subprocess
|
|
import traceback
|
|
import psutil
|
|
from queue import Queue, Empty
|
|
from threading import Thread
|
|
from time import sleep
|
|
|
|
from init import init_bot
|
|
|
|
encode = 'UTF-8'
|
|
|
|
|
|
class RestartBot(Exception):
|
|
pass
|
|
|
|
|
|
def get_pid(name):
|
|
return [p.pid for p in psutil.process_iter() if p.name().find(name) != -1]
|
|
|
|
|
|
def enqueue_output(out, queue):
|
|
for line in iter(out.readline, b''):
|
|
queue.put(line)
|
|
out.close()
|
|
|
|
|
|
init_bot()
|
|
|
|
logging.basicConfig(format="%(msg)s", level=logging.INFO)
|
|
|
|
start = datetime.datetime.now()
|
|
pidlst = []
|
|
|
|
|
|
def run_bot():
|
|
pid_cache = os.path.abspath('.pid_last')
|
|
if os.path.exists(pid_cache):
|
|
with open(pid_cache, 'r') as f:
|
|
pid_last = f.read().split('\n')
|
|
running_pids = get_pid('python')
|
|
print(running_pids)
|
|
print(pid_last)
|
|
for pid in pid_last:
|
|
if int(pid) in running_pids:
|
|
try:
|
|
os.kill(int(pid), 9)
|
|
except (PermissionError, ProcessLookupError):
|
|
pass
|
|
os.remove(pid_cache)
|
|
|
|
botdir = './core/bots/'
|
|
lst = os.listdir(botdir)
|
|
runlst = []
|
|
for x in lst:
|
|
bot = os.path.abspath(f'{botdir}{x}/bot.py')
|
|
if os.path.exists(bot):
|
|
p = subprocess.Popen(['python', bot], shell=False, stdout=subprocess.PIPE, stderr=subprocess.STDOUT,
|
|
cwd=os.path.abspath('.'))
|
|
runlst.append(p)
|
|
pidlst.append(p.pid)
|
|
|
|
with open(pid_cache, 'w') as c:
|
|
c.write('\n'.join(str(p) for p in pidlst))
|
|
|
|
q = Queue()
|
|
threads = []
|
|
for p in runlst:
|
|
threads.append(Thread(target=enqueue_output, args=(p.stdout, q)))
|
|
|
|
for t in threads:
|
|
t.daemon = True
|
|
t.start()
|
|
|
|
while True:
|
|
try:
|
|
line = q.get_nowait()
|
|
except Empty:
|
|
pass
|
|
else:
|
|
try:
|
|
logging.info(line[:-1].decode(encode))
|
|
except Exception:
|
|
print(line)
|
|
traceback.print_exc()
|
|
|
|
# break when all processes are done.
|
|
if all(p.poll() is not None for p in runlst):
|
|
break
|
|
|
|
for p in runlst:
|
|
if p.poll() == 512:
|
|
logging.warning(f'{p.pid} exited with code 512, restart all bots.')
|
|
pidlst.remove(p.pid)
|
|
raise RestartBot
|
|
sleep(0.001)
|
|
|
|
|
|
try:
|
|
while True:
|
|
try:
|
|
run_bot()
|
|
logging.fatal('All bots exited unexpectedly, please check the output')
|
|
break
|
|
except RestartBot:
|
|
for x in pidlst:
|
|
try:
|
|
os.kill(x, 9)
|
|
except (PermissionError, ProcessLookupError):
|
|
pass
|
|
pidlst.clear()
|
|
sleep(5)
|
|
continue
|
|
except KeyboardInterrupt:
|
|
for x in pidlst:
|
|
try:
|
|
os.kill(x, 9)
|
|
except (PermissionError, ProcessLookupError):
|
|
pass
|