alu/tra-superscript
1
0
Fork 0
mirror of https://github.com/titanscouting/tra-superscript.git synced 2024-11-09 22:44:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

delete on start

Browse Source
This commit is contained in:
Dev Singh 2022-03-28 14:43:24 -05:00
parent 25e4babd71
commit ac8002aaf8
2 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
competition/data.py
View File

@ -11,6 +11,7 @@ def pull_new_tba_matches(apikey, competition, cutoff):
for i in json: for i in json:
if i["actual_time"] != None and i["comp_level"] == "qm": if i["actual_time"] != None and i["comp_level"] == "qm":
out.append({"match" : i['match_number'], "blue" : list(map(lambda x: int(x[3:]), i['alliances']['blue']['team_keys'])), "red" : list(map(lambda x: int(x[3:]), i['alliances']['red']['team_keys'])), "winner": i["winning_alliance"]}) out.append({"match" : i['match_number'], "blue" : list(map(lambda x: int(x[3:]), i['alliances']['blue']['team_keys'])), "red" : list(map(lambda x: int(x[3:]), i['alliances']['red']['team_keys'])), "winner": i["winning_alliance"]})
out.sort(key=lambda x: x['match'])
return out return out
def pull_new_tba_matches_manual(apikey, competition, cutoff): def pull_new_tba_matches_manual(apikey, competition, cutoff):
@ -27,6 +28,12 @@ def get_team_match_data(client, competition, team_num):
out[i['match']] = i['data'] out[i['match']] = i['data']
return pd.DataFrame(out) return pd.DataFrame(out)
def clear_metrics(client, competition):
db = client.data_processing
data = db.team_metrics
data.delete_many({competition: competition})
return True
def get_team_pit_data(client, competition, team_num): def get_team_pit_data(client, competition, team_num):
db = client.data_scouting db = client.data_scouting
mdata = db.pitdata mdata = db.pitdata

7
competition/module.py
View File

@ -3,6 +3,7 @@ import data as d
import signal import signal
import numpy as np import numpy as np
from tra_analysis import Analysis as an from tra_analysis import Analysis as an
from tqdm import tqdm
class Module(metaclass = abc.ABCMeta): class Module(metaclass = abc.ABCMeta):
@ -173,6 +174,8 @@ class Metric (Module):
def _process_data(self): def _process_data(self):
d.clear_metrics(self.apikey, self.competition)
elo_N = self.config["tests"]["elo"]["N"] elo_N = self.config["tests"]["elo"]["N"]
elo_K = self.config["tests"]["elo"]["K"] elo_K = self.config["tests"]["elo"]["K"]
@ -180,9 +183,7 @@ class Metric (Module):
red = {} red = {}
blu = {} blu = {}
for match in tqdm(matches, desc="Metrics"):
for match in matches:
red = d.load_metric(self.apikey, self.competition, match, "red", self.config["tests"]) red = d.load_metric(self.apikey, self.competition, match, "red", self.config["tests"])
blu = d.load_metric(self.apikey, self.competition, match, "blue", self.config["tests"]) blu = d.load_metric(self.apikey, self.competition, match, "blue", self.config["tests"])