mirror of
https://github.com/titanscouting/tra-superscript.git
synced 2024-09-20 00:47:56 +00:00
unified main_lin, main_win functions
This commit is contained in:
Arthur Lu
parent
e9258c831d
commit
342dae3022
@ -205,7 +205,223 @@ sample_json = """{
|
|||||||
}
|
}
|
||||||
}"""
|
}"""
|
||||||
|
|
||||||
def main_lin(pid_path):
|
def main(send, verbose = False):
|
||||||
|
|
||||||
|
if verbose :
|
||||||
|
|
||||||
|
warnings.filterwarnings("ignore")
|
||||||
|
sys.stderr = open("errorlog.txt", "w")
|
||||||
|
|
||||||
|
splash(__version__)
|
||||||
|
|
||||||
|
loop_exit_code = 0
|
||||||
|
loop_stored_exception = None
|
||||||
|
|
||||||
|
while True:
|
||||||
|
|
||||||
|
try:
|
||||||
|
|
||||||
|
loop_start = time.time()
|
||||||
|
|
||||||
|
current_time = time.time()
|
||||||
|
send(stdout, INF, "current time: " + str(current_time))
|
||||||
|
|
||||||
|
send(stdout, INF, "loading config at <" + config_path + ">", code = 0)
|
||||||
|
|
||||||
|
config = {}
|
||||||
|
if load_config(config_path, config) == 1:
|
||||||
|
send(stderr, ERR, "could not find config at <" + config_path + ">, generating blank config and exiting", code = 100)
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
send(stdout, INF, "found and opened config at <" + config_path + ">", code = 0)
|
||||||
|
|
||||||
|
error_flag = False
|
||||||
|
|
||||||
|
try:
|
||||||
|
competition = config["competition"]
|
||||||
|
except:
|
||||||
|
send(stderr, ERR, "could not find competition field in config", code = 101)
|
||||||
|
error_flag = True
|
||||||
|
try:
|
||||||
|
match_tests = config["statistics"]["match"]
|
||||||
|
except:
|
||||||
|
send(stderr, ERR, "could not find match_tests field in config", code = 102)
|
||||||
|
error_flag = True
|
||||||
|
try:
|
||||||
|
metrics_tests = config["statistics"]["metric"]
|
||||||
|
except:
|
||||||
|
send(stderr, ERR, "could not find metrics_tests field in config", code = 103)
|
||||||
|
error_flag = True
|
||||||
|
try:
|
||||||
|
pit_tests = config["statistics"]["pit"]
|
||||||
|
except:
|
||||||
|
send(stderr, ERR, "could not find pit_tests field in config", code = 104)
|
||||||
|
error_flag = True
|
||||||
|
|
||||||
|
if error_flag:
|
||||||
|
sys.exit(1)
|
||||||
|
error_flag = False
|
||||||
|
|
||||||
|
if competition == None or competition == "":
|
||||||
|
send(stderr, ERR, "competition field in config must not be empty", code = 105)
|
||||||
|
error_flag = True
|
||||||
|
if match_tests == None:
|
||||||
|
send(stderr, ERR, "match_tests field in config must not be empty", code = 106)
|
||||||
|
error_flag = True
|
||||||
|
if metrics_tests == None:
|
||||||
|
send(stderr, ERR, "metrics_tests field in config must not be empty", code = 107)
|
||||||
|
error_flag = True
|
||||||
|
if pit_tests == None:
|
||||||
|
send(stderr, ERR, "pit_tests field in config must not be empty", code = 108)
|
||||||
|
error_flag = True
|
||||||
|
|
||||||
|
if error_flag:
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
send(stdout, INF, "found and loaded competition, match_tests, metrics_tests, pit_tests from config")
|
||||||
|
|
||||||
|
sys_max_threads = os.cpu_count()
|
||||||
|
try:
|
||||||
|
cfg_max_threads = config["max-threads"]
|
||||||
|
except:
|
||||||
|
send(stderr, ERR, "max-threads field in config must not be empty, refer to documentation for configuration options", code = 109)
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
if cfg_max_threads > -sys_max_threads and cfg_max_threads < 0 :
|
||||||
|
alloc_processes = sys_max_threads + cfg_max_threads
|
||||||
|
elif cfg_max_threads > 0 and cfg_max_threads < 1:
|
||||||
|
alloc_processes = math.floor(cfg_max_threads * sys_max_threads)
|
||||||
|
elif cfg_max_threads > 1 and cfg_max_threads <= sys_max_threads:
|
||||||
|
alloc_processes = cfg_max_threads
|
||||||
|
elif cfg_max_threads == 0:
|
||||||
|
alloc_processes = sys_max_threads
|
||||||
|
else:
|
||||||
|
send(stderr, ERR, "max-threads must be between -" + str(sys_max_threads) + " and " + str(sys_max_threads) + ", but got " + cfg_max_threads, code = 110)
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
send(stdout, INF, "found and loaded max-threads from config")
|
||||||
|
send(stdout, INF, "attempting to start " + str(alloc_processes) + " threads")
|
||||||
|
try:
|
||||||
|
exec_threads = Pool(processes = alloc_processes)
|
||||||
|
except Exception as e:
|
||||||
|
send(stderr, ERR, "unable to start threads", code = 200)
|
||||||
|
send(stderr, INF, e)
|
||||||
|
sys.exit(1)
|
||||||
|
send(stdout, INF, "successfully initialized " + str(alloc_processes) + " threads")
|
||||||
|
|
||||||
|
exit_flag = False
|
||||||
|
|
||||||
|
try:
|
||||||
|
apikey = config["key"]["database"]
|
||||||
|
except:
|
||||||
|
send(stderr, ERR, "database key field in config must be present", code = 111)
|
||||||
|
exit_flag = True
|
||||||
|
try:
|
||||||
|
tbakey = config["key"]["tba"]
|
||||||
|
except:
|
||||||
|
send(stderr, ERR, "tba key field in config must be present", code = 112)
|
||||||
|
exit_flag = True
|
||||||
|
|
||||||
|
if apikey == None or apikey == "":
|
||||||
|
send(stderr, ERR, "database key field in config must not be empty, please populate the database key")
|
||||||
|
exit_flag = True
|
||||||
|
if tbakey == None or tbakey == "":
|
||||||
|
send(stderr, ERR, "tba key field in config must not be empty, please populate the tba key")
|
||||||
|
exit_flag = True
|
||||||
|
|
||||||
|
if exit_flag:
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
send(stdout, INF, "found and loaded database and tba keys")
|
||||||
|
|
||||||
|
client = pymongo.MongoClient(apikey)
|
||||||
|
|
||||||
|
previous_time = get_previous_time(client)
|
||||||
|
send(stdout, INF, "analysis backtimed to: " + str(previous_time))
|
||||||
|
|
||||||
|
start = time.time()
|
||||||
|
send(stdout, INF, "loading match data")
|
||||||
|
match_data = load_match(client, competition)
|
||||||
|
send(stdout, INF, "finished loading match data in " + str(time.time() - start) + " seconds")
|
||||||
|
|
||||||
|
start = time.time()
|
||||||
|
send(stdout, INF, "performing analysis on match data")
|
||||||
|
results = matchloop(client, competition, match_data, match_tests, exec_threads)
|
||||||
|
send(stdout, INF, "finished match analysis in " + str(time.time() - start) + " seconds")
|
||||||
|
|
||||||
|
start = time.time()
|
||||||
|
send(stdout, INF, "uploading match results to database")
|
||||||
|
push_match(client, competition, results)
|
||||||
|
send(stdout, INF, "finished uploading match results in " + str(time.time() - start) + " seconds")
|
||||||
|
|
||||||
|
start = time.time()
|
||||||
|
send(stdout, INF, "performing analysis on team metrics")
|
||||||
|
results = metricloop(tbakey, client, competition, current_time, metrics_tests)
|
||||||
|
send(stdout, INF, "finished metric analysis and pushed to database in " + str(time.time() - start) + " seconds")
|
||||||
|
|
||||||
|
start = time.time()
|
||||||
|
send(stdout, INF, "loading pit data")
|
||||||
|
pit_data = load_pit(client, competition)
|
||||||
|
send(stdout, INF, "finished loading pit data in " + str(time.time() - start) + " seconds")
|
||||||
|
|
||||||
|
start = time.time()
|
||||||
|
send(stdout, INF, "performing analysis on pit data")
|
||||||
|
results = pitloop(client, competition, pit_data, pit_tests)
|
||||||
|
send(stdout, INF, "finished pit analysis in " + str(time.time() - start) + " seconds")
|
||||||
|
|
||||||
|
start = time.time()
|
||||||
|
send(stdout, INF, "uploading pit results to database")
|
||||||
|
push_pit(client, competition, results)
|
||||||
|
send(stdout, INF, "finished uploading pit results in " + str(time.time() - start) + " seconds")
|
||||||
|
|
||||||
|
client.close()
|
||||||
|
|
||||||
|
set_current_time(client, current_time)
|
||||||
|
send(stdout, INF, "finished all tests in " + str(time.time() - loop_start) + " seconds, looping")
|
||||||
|
|
||||||
|
except KeyboardInterrupt:
|
||||||
|
send(stdout, INF, "detected KeyboardInterrupt, killing threads")
|
||||||
|
if "exec_threads" in locals():
|
||||||
|
exec_threads.terminate()
|
||||||
|
exec_threads.join()
|
||||||
|
exec_threads.close()
|
||||||
|
send(stdout, INF, "terminated threads, exiting")
|
||||||
|
loop_stored_exception = sys.exc_info()
|
||||||
|
loop_exit_code = 0
|
||||||
|
break
|
||||||
|
except Exception as e:
|
||||||
|
send(stderr, ERR, "encountered an exception while running")
|
||||||
|
print(e, file = stderr)
|
||||||
|
loop_exit_code = 1
|
||||||
|
break
|
||||||
|
|
||||||
|
sys.exit(loop_exit_code)
|
||||||
|
|
||||||
|
def load_config(path, config_vector):
|
||||||
|
try:
|
||||||
|
f = open(path, "r")
|
||||||
|
config_vector.update(json.load(f))
|
||||||
|
f.close()
|
||||||
|
return 0
|
||||||
|
except:
|
||||||
|
f = open(path, "w")
|
||||||
|
f.write(sample_json)
|
||||||
|
f.close()
|
||||||
|
return 1
|
||||||
|
|
||||||
|
def save_config(path, config_vector):
|
||||||
|
try:
|
||||||
|
f = open(path)
|
||||||
|
json.dump(config_vector)
|
||||||
|
f.close()
|
||||||
|
return 0
|
||||||
|
except:
|
||||||
|
return 1
|
||||||
|
|
||||||
|
def start(pid_path, verbose = False):
|
||||||
|
|
||||||
|
if not verbose:
|
||||||
|
|
||||||
f = open('errorlog.txt', 'w+')
|
f = open('errorlog.txt', 'w+')
|
||||||
with daemon.DaemonContext(
|
with daemon.DaemonContext(
|
||||||
working_directory=os.getcwd(),
|
working_directory=os.getcwd(),
|
||||||
@ -224,403 +440,28 @@ def main_lin(pid_path):
|
|||||||
clients.remove(client)
|
clients.remove(client)
|
||||||
break
|
break
|
||||||
|
|
||||||
|
async def send_one(client, data):
|
||||||
|
await client.send(data)
|
||||||
|
|
||||||
|
def send(target, level, message, code = 0):
|
||||||
|
message_clients = clients.copy()
|
||||||
|
for client in message_clients:
|
||||||
|
try:
|
||||||
|
asyncio.run(send_one(client, message))
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
|
||||||
clients = []
|
clients = []
|
||||||
start_server = websockets.serve(handler, "0.0.0.0", 5678)
|
start_server = websockets.serve(handler, "0.0.0.0", 5678)
|
||||||
|
|
||||||
asyncio.get_event_loop().run_until_complete(start_server)
|
asyncio.get_event_loop().run_until_complete(start_server)
|
||||||
threading.Thread(target = asyncio.get_event_loop().run_forever).start()
|
threading.Thread(target = asyncio.get_event_loop().run_forever).start()
|
||||||
|
|
||||||
while True:
|
main(send)
|
||||||
|
|
||||||
async def send_one(client, data):
|
|
||||||
await client.send(data)
|
|
||||||
|
|
||||||
def send(data):
|
|
||||||
message_clients = clients.copy()
|
|
||||||
for client in message_clients:
|
|
||||||
try:
|
|
||||||
asyncio.run(send_one(client, data))
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
|
|
||||||
try:
|
|
||||||
|
|
||||||
loop_start = time.time()
|
|
||||||
|
|
||||||
current_time = time.time()
|
|
||||||
send("current time: " + str(current_time))
|
|
||||||
|
|
||||||
config = {}
|
|
||||||
if load_config(config_path, config) == 1:
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
error_flag = False
|
|
||||||
|
|
||||||
try:
|
|
||||||
competition = config["competition"]
|
|
||||||
except:
|
|
||||||
send("could not find competition field in config")
|
|
||||||
error_flag = True
|
|
||||||
try:
|
|
||||||
match_tests = config["statistics"]["match"]
|
|
||||||
except:
|
|
||||||
send("could not find match_tests field in config")
|
|
||||||
error_flag = True
|
|
||||||
try:
|
|
||||||
metrics_tests = config["statistics"]["metric"]
|
|
||||||
except:
|
|
||||||
send("could not find metrics_tests field in config")
|
|
||||||
error_flag = True
|
|
||||||
try:
|
|
||||||
pit_tests = config["statistics"]["pit"]
|
|
||||||
except:
|
|
||||||
send("could not find pit_tests field in config")
|
|
||||||
error_flag = True
|
|
||||||
|
|
||||||
if error_flag:
|
|
||||||
sys.exit(1)
|
|
||||||
error_flag = False
|
|
||||||
|
|
||||||
if competition == None or competition == "":
|
|
||||||
send("competition field in config must not be empty")
|
|
||||||
error_flag = True
|
|
||||||
if match_tests == None:
|
|
||||||
send("match_tests field in config must not be empty")
|
|
||||||
error_flag = True
|
|
||||||
if metrics_tests == None:
|
|
||||||
send("metrics_tests field in config must not be empty")
|
|
||||||
error_flag = True
|
|
||||||
if pit_tests == None:
|
|
||||||
send("pit_tests field in config must not be empty")
|
|
||||||
error_flag = True
|
|
||||||
|
|
||||||
if error_flag:
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
send("found and loaded competition, match_tests, metrics_tests, pit_tests from config")
|
|
||||||
|
|
||||||
sys_max_threads = os.cpu_count()
|
|
||||||
try:
|
|
||||||
cfg_max_threads = config["max-threads"]
|
|
||||||
except:
|
|
||||||
send("max-threads field in config must not be empty, refer to documentation for configuration options", code = 109)
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
if cfg_max_threads > -sys_max_threads and cfg_max_threads < 0 :
|
|
||||||
alloc_processes = sys_max_threads + cfg_max_threads
|
|
||||||
elif cfg_max_threads > 0 and cfg_max_threads < 1:
|
|
||||||
alloc_processes = math.floor(cfg_max_threads * sys_max_threads)
|
|
||||||
elif cfg_max_threads > 1 and cfg_max_threads <= sys_max_threads:
|
|
||||||
alloc_processes = cfg_max_threads
|
|
||||||
elif cfg_max_threads == 0:
|
|
||||||
alloc_processes = sys_max_threads
|
|
||||||
else:
|
else:
|
||||||
send("max-threads must be between -" + str(sys_max_threads) + " and " + str(sys_max_threads) + ", but got " + cfg_max_threads)
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
send("found and loaded max-threads from config")
|
main(log, verbose=verbose)
|
||||||
send("attempting to start " + str(alloc_processes) + " threads")
|
|
||||||
try:
|
|
||||||
exec_threads = Pool(processes = alloc_processes)
|
|
||||||
except Exception as e:
|
|
||||||
send("unable to start threads")
|
|
||||||
sys.exit(1)
|
|
||||||
send("successfully initialized " + str(alloc_processes) + " threads")
|
|
||||||
|
|
||||||
exit_flag = False
|
|
||||||
|
|
||||||
try:
|
|
||||||
apikey = config["key"]["database"]
|
|
||||||
except:
|
|
||||||
send("database key field in config must be present")
|
|
||||||
exit_flag = True
|
|
||||||
try:
|
|
||||||
tbakey = config["key"]["tba"]
|
|
||||||
except:
|
|
||||||
send("tba key field in config must be present")
|
|
||||||
exit_flag = True
|
|
||||||
|
|
||||||
if apikey == None or apikey == "":
|
|
||||||
send("database key field in config must not be empty, please populate the database key")
|
|
||||||
exit_flag = True
|
|
||||||
if tbakey == None or tbakey == "":
|
|
||||||
send("tba key field in config must not be empty, please populate the tba key")
|
|
||||||
exit_flag = True
|
|
||||||
|
|
||||||
if exit_flag:
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
send("found and loaded database and tba keys")
|
|
||||||
|
|
||||||
client = pymongo.MongoClient(apikey)
|
|
||||||
|
|
||||||
previous_time = get_previous_time(client)
|
|
||||||
send("analysis backtimed to: " + str(previous_time))
|
|
||||||
|
|
||||||
start = time.time()
|
|
||||||
send("loading match data")
|
|
||||||
match_data = load_match(client, competition)
|
|
||||||
send("finished loading match data in " + str(time.time() - start) + " seconds")
|
|
||||||
|
|
||||||
start = time.time()
|
|
||||||
send("performing analysis on match data")
|
|
||||||
results = matchloop(client, competition, match_data, match_tests, exec_threads)
|
|
||||||
send("finished match analysis in " + str(time.time() - start) + " seconds")
|
|
||||||
|
|
||||||
start = time.time()
|
|
||||||
send("uploading match results to database")
|
|
||||||
push_match(client, competition, results)
|
|
||||||
send("finished uploading match results in " + str(time.time() - start) + " seconds")
|
|
||||||
|
|
||||||
start = time.time()
|
|
||||||
send("performing analysis on team metrics")
|
|
||||||
results = metricloop(tbakey, client, competition, current_time, metrics_tests)
|
|
||||||
send("finished metric analysis and pushed to database in " + str(time.time() - start) + " seconds")
|
|
||||||
|
|
||||||
start = time.time()
|
|
||||||
send("loading pit data")
|
|
||||||
pit_data = load_pit(client, competition)
|
|
||||||
send("finished loading pit data in " + str(time.time() - start) + " seconds")
|
|
||||||
|
|
||||||
start = time.time()
|
|
||||||
send("performing analysis on pit data")
|
|
||||||
results = pitloop(client, competition, pit_data, pit_tests)
|
|
||||||
send("finished pit analysis in " + str(time.time() - start) + " seconds")
|
|
||||||
|
|
||||||
start = time.time()
|
|
||||||
send("uploading pit results to database")
|
|
||||||
push_pit(client, competition, results)
|
|
||||||
send("finished uploading pit results in " + str(time.time() - start) + " seconds")
|
|
||||||
|
|
||||||
set_current_time(client, current_time)
|
|
||||||
send("finished all tests in " + str(time.time() - loop_start) + " seconds, looping")
|
|
||||||
|
|
||||||
except KeyboardInterrupt:
|
|
||||||
send("detected KeyboardInterrupt, killing threads")
|
|
||||||
if "exec_threads" in locals():
|
|
||||||
exec_threads.terminate()
|
|
||||||
exec_threads.join()
|
|
||||||
exec_threads.close()
|
|
||||||
send("terminated threads, exiting")
|
|
||||||
loop_stored_exception = sys.exc_info()
|
|
||||||
loop_exit_code = 0
|
|
||||||
break
|
|
||||||
except Exception as e:
|
|
||||||
send("encountered an exception while running")
|
|
||||||
send(str(e))
|
|
||||||
loop_exit_code = 1
|
|
||||||
break
|
|
||||||
|
|
||||||
sys.exit(loop_exit_code)
|
|
||||||
|
|
||||||
def main_win():
|
|
||||||
|
|
||||||
warnings.filterwarnings("ignore")
|
|
||||||
sys.stderr = open("errorlog.txt", "w")
|
|
||||||
|
|
||||||
splash(__version__)
|
|
||||||
|
|
||||||
loop_exit_code = 0
|
|
||||||
loop_stored_exception = None
|
|
||||||
|
|
||||||
while True:
|
|
||||||
|
|
||||||
try:
|
|
||||||
|
|
||||||
loop_start = time.time()
|
|
||||||
|
|
||||||
current_time = time.time()
|
|
||||||
log(stdout, INF, "current time: " + str(current_time))
|
|
||||||
|
|
||||||
config = {}
|
|
||||||
if load_config(config_path, config) == 1:
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
error_flag = False
|
|
||||||
|
|
||||||
try:
|
|
||||||
competition = config["competition"]
|
|
||||||
except:
|
|
||||||
log(stderr, ERR, "could not find competition field in config", code = 101)
|
|
||||||
error_flag = True
|
|
||||||
try:
|
|
||||||
match_tests = config["statistics"]["match"]
|
|
||||||
except:
|
|
||||||
log(stderr, ERR, "could not find match_tests field in config", code = 102)
|
|
||||||
error_flag = True
|
|
||||||
try:
|
|
||||||
metrics_tests = config["statistics"]["metric"]
|
|
||||||
except:
|
|
||||||
log(stderr, ERR, "could not find metrics_tests field in config", code = 103)
|
|
||||||
error_flag = True
|
|
||||||
try:
|
|
||||||
pit_tests = config["statistics"]["pit"]
|
|
||||||
except:
|
|
||||||
log(stderr, ERR, "could not find pit_tests field in config", code = 104)
|
|
||||||
error_flag = True
|
|
||||||
|
|
||||||
if error_flag:
|
|
||||||
sys.exit(1)
|
|
||||||
error_flag = False
|
|
||||||
|
|
||||||
if competition == None or competition == "":
|
|
||||||
log(stderr, ERR, "competition field in config must not be empty", code = 105)
|
|
||||||
error_flag = True
|
|
||||||
if match_tests == None:
|
|
||||||
log(stderr, ERR, "match_tests field in config must not be empty", code = 106)
|
|
||||||
error_flag = True
|
|
||||||
if metrics_tests == None:
|
|
||||||
log(stderr, ERR, "metrics_tests field in config must not be empty", code = 107)
|
|
||||||
error_flag = True
|
|
||||||
if pit_tests == None:
|
|
||||||
log(stderr, ERR, "pit_tests field in config must not be empty", code = 108)
|
|
||||||
error_flag = True
|
|
||||||
|
|
||||||
if error_flag:
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
log(stdout, INF, "found and loaded competition, match_tests, metrics_tests, pit_tests from config")
|
|
||||||
|
|
||||||
sys_max_threads = os.cpu_count()
|
|
||||||
try:
|
|
||||||
cfg_max_threads = config["max-threads"]
|
|
||||||
except:
|
|
||||||
log(stderr, ERR, "max-threads field in config must not be empty, refer to documentation for configuration options", code = 109)
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
if cfg_max_threads > -sys_max_threads and cfg_max_threads < 0 :
|
|
||||||
alloc_processes = sys_max_threads + cfg_max_threads
|
|
||||||
elif cfg_max_threads > 0 and cfg_max_threads < 1:
|
|
||||||
alloc_processes = math.floor(cfg_max_threads * sys_max_threads)
|
|
||||||
elif cfg_max_threads > 1 and cfg_max_threads <= sys_max_threads:
|
|
||||||
alloc_processes = cfg_max_threads
|
|
||||||
elif cfg_max_threads == 0:
|
|
||||||
alloc_processes = sys_max_threads
|
|
||||||
else:
|
|
||||||
log(stderr, ERR, "max-threads must be between -" + str(sys_max_threads) + " and " + str(sys_max_threads) + ", but got " + cfg_max_threads, code = 110)
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
log(stdout, INF, "found and loaded max-threads from config")
|
|
||||||
log(stdout, INF, "attempting to start " + str(alloc_processes) + " threads")
|
|
||||||
try:
|
|
||||||
exec_threads = Pool(processes = alloc_processes)
|
|
||||||
except Exception as e:
|
|
||||||
log(stderr, ERR, "unable to start threads", code = 200)
|
|
||||||
log(stderr, INF, e)
|
|
||||||
sys.exit(1)
|
|
||||||
log(stdout, INF, "successfully initialized " + str(alloc_processes) + " threads")
|
|
||||||
|
|
||||||
exit_flag = False
|
|
||||||
|
|
||||||
try:
|
|
||||||
apikey = config["key"]["database"]
|
|
||||||
except:
|
|
||||||
log(stderr, ERR, "database key field in config must be present", code = 111)
|
|
||||||
exit_flag = True
|
|
||||||
try:
|
|
||||||
tbakey = config["key"]["tba"]
|
|
||||||
except:
|
|
||||||
log(stderr, ERR, "tba key field in config must be present", code = 112)
|
|
||||||
exit_flag = True
|
|
||||||
|
|
||||||
if apikey == None or apikey == "":
|
|
||||||
log(stderr, ERR, "database key field in config must not be empty, please populate the database key")
|
|
||||||
exit_flag = True
|
|
||||||
if tbakey == None or tbakey == "":
|
|
||||||
log(stderr, ERR, "tba key field in config must not be empty, please populate the tba key")
|
|
||||||
exit_flag = True
|
|
||||||
|
|
||||||
if exit_flag:
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
log(stdout, INF, "found and loaded database and tba keys")
|
|
||||||
|
|
||||||
client = pymongo.MongoClient(apikey)
|
|
||||||
|
|
||||||
previous_time = get_previous_time(client)
|
|
||||||
log(stdout, INF, "analysis backtimed to: " + str(previous_time))
|
|
||||||
|
|
||||||
start = time.time()
|
|
||||||
log(stdout, INF, "loading match data")
|
|
||||||
match_data = load_match(client, competition)
|
|
||||||
log(stdout, INF, "finished loading match data in " + str(time.time() - start) + " seconds")
|
|
||||||
|
|
||||||
start = time.time()
|
|
||||||
log(stdout, INF, "performing analysis on match data")
|
|
||||||
results = matchloop(client, competition, match_data, match_tests, exec_threads)
|
|
||||||
log(stdout, INF, "finished match analysis in " + str(time.time() - start) + " seconds")
|
|
||||||
|
|
||||||
start = time.time()
|
|
||||||
log(stdout, INF, "uploading match results to database")
|
|
||||||
push_match(client, competition, results)
|
|
||||||
log(stdout, INF, "finished uploading match results in " + str(time.time() - start) + " seconds")
|
|
||||||
|
|
||||||
start = time.time()
|
|
||||||
log(stdout, INF, "performing analysis on team metrics")
|
|
||||||
results = metricloop(tbakey, client, competition, current_time, metrics_tests)
|
|
||||||
log(stdout, INF, "finished metric analysis and pushed to database in " + str(time.time() - start) + " seconds")
|
|
||||||
|
|
||||||
start = time.time()
|
|
||||||
log(stdout, INF, "loading pit data")
|
|
||||||
pit_data = load_pit(client, competition)
|
|
||||||
log(stdout, INF, "finished loading pit data in " + str(time.time() - start) + " seconds")
|
|
||||||
|
|
||||||
start = time.time()
|
|
||||||
log(stdout, INF, "performing analysis on pit data")
|
|
||||||
results = pitloop(client, competition, pit_data, pit_tests)
|
|
||||||
log(stdout, INF, "finished pit analysis in " + str(time.time() - start) + " seconds")
|
|
||||||
|
|
||||||
start = time.time()
|
|
||||||
log(stdout, INF, "uploading pit results to database")
|
|
||||||
push_pit(client, competition, results)
|
|
||||||
log(stdout, INF, "finished uploading pit results in " + str(time.time() - start) + " seconds")
|
|
||||||
|
|
||||||
set_current_time(client, current_time)
|
|
||||||
log(stdout, INF, "finished all tests in " + str(time.time() - loop_start) + " seconds, looping")
|
|
||||||
|
|
||||||
except KeyboardInterrupt:
|
|
||||||
log(stdout, INF, "detected KeyboardInterrupt, killing threads")
|
|
||||||
if "exec_threads" in locals():
|
|
||||||
exec_threads.terminate()
|
|
||||||
exec_threads.join()
|
|
||||||
exec_threads.close()
|
|
||||||
log(stdout, INF, "terminated threads, exiting")
|
|
||||||
loop_stored_exception = sys.exc_info()
|
|
||||||
loop_exit_code = 0
|
|
||||||
break
|
|
||||||
except Exception as e:
|
|
||||||
log(stderr, ERR, "encountered an exception while running")
|
|
||||||
print(e, file = stderr)
|
|
||||||
loop_exit_code = 1
|
|
||||||
break
|
|
||||||
|
|
||||||
sys.exit(loop_exit_code)
|
|
||||||
|
|
||||||
def load_config(path, config_vector):
|
|
||||||
try:
|
|
||||||
f = open(path, "r")
|
|
||||||
config_vector.update(json.load(f))
|
|
||||||
f.close()
|
|
||||||
#socket.send("found and opened config at <" + path + ">")
|
|
||||||
return 0
|
|
||||||
except:
|
|
||||||
#log(stderr, ERR, "could not find config at <" + path + ">, generating blank config and exiting", code = 100)
|
|
||||||
f = open(path, "w")
|
|
||||||
f.write(sample_json)
|
|
||||||
f.close()
|
|
||||||
return 1
|
|
||||||
|
|
||||||
def save_config(path, config_vector):
|
|
||||||
try:
|
|
||||||
f = open(path)
|
|
||||||
json.dump(config_vector)
|
|
||||||
f.close()
|
|
||||||
return 0
|
|
||||||
except:
|
|
||||||
return 1
|
|
||||||
|
|
||||||
def start(pid_path):
|
|
||||||
main_lin(pid_path)
|
|
||||||
|
|
||||||
def stop(pid_path):
|
def stop(pid_path):
|
||||||
try:
|
try:
|
||||||
@ -652,7 +493,7 @@ if __name__ == "__main__":
|
|||||||
|
|
||||||
if sys.platform.startswith("win"):
|
if sys.platform.startswith("win"):
|
||||||
freeze_support()
|
freeze_support()
|
||||||
main_win()
|
start(None, verbose = True)
|
||||||
|
|
||||||
else:
|
else:
|
||||||
import daemon
|
import daemon
|
||||||
@ -667,7 +508,7 @@ if __name__ == "__main__":
|
|||||||
elif 'restart' == sys.argv[1]:
|
elif 'restart' == sys.argv[1]:
|
||||||
restart(pid_path)
|
restart(pid_path)
|
||||||
elif 'verbose' == sys.argv[1]:
|
elif 'verbose' == sys.argv[1]:
|
||||||
main_win()
|
start(None, verbose = True)
|
||||||
else:
|
else:
|
||||||
print("usage: %s start|stop|restart|verbose" % sys.argv[0])
|
print("usage: %s start|stop|restart|verbose" % sys.argv[0])
|
||||||
sys.exit(2)
|
sys.exit(2)
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user