readded tra.py as a fallback option

made changes to tra-cli.py

Signed-off-by: Arthur Lu <learthurgo@gmail.com>

data-analysis/tra-cli.py

@@ -1,9 +1,12 @@
 import argparse
 from tasks import Tasker
+import test
+import threading
+from multiprocessing import Process, Queue
 
 t = Tasker()
 
-task_map = {"match":None, "metric":None, "pit":None}
+task_map = {"match":None, "metric":None, "pit":None, "test":None}
 status_map = {"match":None, "metric":None, "pit":None}
 status_map.update(task_map)
 
@@ -22,4 +25,9 @@ parser_status = subparsers.add_parser("status", help = "//stophelp//")
 parser_status.add_argument("targets", metavar = "T", nargs = "*", choices = status_map.keys())
 parser_status.set_defaults(which = "status")
 
-args = parser.parse_args()
+args = parser.parse_args()
+
+if(args.which == "start" and "test" in args.targets):
+	a = test.testcls()
+	tmain = Process(name = "main", target = a.main)
+	tmain.start()

data-analysis/tra.py

@@ -0,0 +1,166 @@
+import json
+import superscript as su
+import threading
+
+__author__ = (
+	"Arthur Lu <learthurgo@gmail.com>",
+)
+
+match_ = False
+metric_ = False
+pit_ = False
+
+match_enable = True
+metric_enable = True
+pit_enable = True
+
+config = {}
+
+def __init__(self):
+
+	global match_
+	global metric_
+	global pit_
+
+	global match_enable
+	global metric_enable
+	global pit_enable
+
+	config = su.load_config("config.json")
+
+def match(self):
+
+	match_ = True
+
+	apikey = config["key"]["database"]
+	competition = config["competition"]
+	tests = config["statistics"]["match"]
+
+	data = su.load_match(apikey, competition)
+	su.matchloop(apikey, competition, data, tests)
+
+	match_ = False
+
+	if match_enable == True and match_ == False:
+		
+		task = threading.Thread(name = "match", target = match)
+		task.start()
+
+def metric():
+
+	metric_ = True
+
+	apikey = config["key"]["database"]
+	tbakey = config["key"]["tba"]
+	competition = config["competition"]
+	metric = config["statistics"]["metric"]
+
+	timestamp = su.get_previous_time(apikey)
+
+	su.metricloop(tbakey, apikey, competition, timestamp, metric)
+
+	metric_ = False
+
+	if metric_enable == True and metric_ == False:
+		
+		task = threading.Thread(name = "match", target = metric)
+		task.start()
+
+def pit():
+
+	pit_ = True
+
+	apikey = config["key"]["database"]
+	competition = config["competition"]
+	tests = config["statistics"]["pit"]
+
+	data = su.load_pit(apikey, competition)
+	su.pitloop(apikey, competition, data, tests)
+
+	pit_ = False
+
+	if pit_enable == True and pit_ == False:
+		
+		task = threading.Thread(name = "pit", target = pit)
+		task.start()
+
+def start_match():
+	task = threading.Thread(name = "match", target = match)
+	task.start()
+
+def start_metric():
+	task = threading.Thread(name = "match", target = metric)
+	task.start()
+
+def start_pit():
+	task = threading.Thread(name = "pit", target = pit)
+	task.start()
+
+def main():
+
+	init()
+	start_match()
+	start_metric()
+	start_pit()
+
+	exit = False
+	while(not exit):
+
+		i = input("> ")
+		cmds = i.split(" ")
+		cmds = [x for x in cmds if x != ""]
+		l = len(cmds)
+
+		if(l == 0):
+			pass
+		else:
+			if(cmds[0] == "exit"):
+				if(l == 1):
+					exit = True
+				else:
+					print("exit command expected no arguments but encountered " + str(l - 1))
+			if(cmds[0] == "status"):
+				if(l == 1):
+					print("status command expected 1 argument but encountered none\ntype status help for usage")
+				elif(l > 2):
+					print("status command expected 1 argument but encountered " + str(l - 1))
+				elif(cmds[1] == "threads"):
+					threads = threading.enumerate()
+					threads = [x.getName() for x in threads]
+					print("running threads:")
+					for thread in threads:
+						print("    " + thread)
+				elif(cmds[1] == "flags"):
+					print("current flags:")
+					print("    match running: " + match_)
+					print("    metric running: " + metric_)
+					print("    pit running: " + pit_)
+					print("    match enable: " + match_enable)
+					print("    metric enable: " + metric_enable)
+					print("    pit enable: " + pit_enable)
+				elif(cmds[1] == "config"):
+					print("current config:")
+					print(json.dumps(config))
+				elif(cmds[1] == "all"):
+					threads = threading.enumerate()
+					threads = [x.getName() for x in threads]
+					print("running threads:")
+					for thread in threads:
+						print("    " + thread)
+					print("current flags:")
+					print("    match running: " + match_)
+					print("    metric running: " + metric_)
+					print("    pit running: " + pit_)
+					print("    match enable: " + match_enable)
+					print("    metric enable: " + metric_enable)
+					print("    pit enable: " + pit_enable)
+				elif(cmds[1] == "help"):
+					print("usage: status [arg]\nDisplays the status of the tra data analysis threads.\nArguments:\n    threads - prints the stuatus ofcurrently running threads\n    flags - prints the status of control and indicator flags\n    config - prints the current configuration information\n    all - prints all statuses\n    <name_of_thread> - prints the status of a specific thread")
+				else:
+					threads = threading.enumerate()
+					threads = [x.getName() for x in threads]
+					if(cmds[1] in threads):
+						print(cmds[1] + " is running")
+				
+if(__name__ == "__main__"):
+	main()