diff --git a/data analysis/get_team_rankings.py b/data analysis/get_team_rankings.py new file mode 100644 index 00000000..cec2aa08 --- /dev/null +++ b/data analysis/get_team_rankings.py @@ -0,0 +1,59 @@ +import data as d +from analysis import analysis as an +import pymongo +import operator + +def load_config(file): + config_vector = {} + file = an.load_csv(file) + for line in file[1:]: + config_vector[line[0]] = line[1:] + + return (file[0][0], config_vector) + +def get_metrics_processed_formatted(apikey, competition): + client = pymongo.MongoClient(apikey) + db = client.data_scouting + mdata = db.teamlist + x=mdata.find_one({"competition":competition}) + out = {} + for i in x: + try: + out[int(i)] = d.get_team_metrics_data(apikey, competition, int(i)) + except: + pass + return out + +def main(): + + apikey = an.load_csv("keys.txt")[0][0] + tbakey = an.load_csv("keys.txt")[1][0] + + competition, config = load_config("config.csv") + + metrics = get_metrics_processed_formatted(apikey, competition) + + elo = {} + gl2 = {} + + for team in metrics: + + elo[team] = metrics[team]["metrics"]["elo"]["score"] + gl2[team] = metrics[team]["metrics"]["gl2"]["score"] + + elo = {k: v for k, v in sorted(elo.items(), key=lambda item: item[1])} + gl2 = {k: v for k, v in sorted(gl2.items(), key=lambda item: item[1])} + + for team in elo: + + print("teams sorted by elo:") + print("" + str(team) + " | " + str(elo[team])) + + print("*"*25) + + for team in gl2: + + print("teams sorted by glicko2:") + print("" + str(team) + " | " + str(gl2[team])) + +main() \ No newline at end of file