import json
from pprint import pprint

with open('rank.json') as f:
    data = json.load(f)

kek =  (sorted(data, key=lambda d: d["gpa"], reverse=True))

total = 0
moy = 0.0

obj = {
    "rank Adrien" : 0,
    "rank Hugo" : 0,
    "box" :
        {'BDX' : {
            "Moyen":0.0,
            "Total":0
            },
        'LIL' : {
            "Moyen":0.0,
            "Total":0
            },
        'LYN' : {
            "Moyen":0.0,
            "Total":0
            },
        'MAR' : {
            "Moyen":0.0,
            "Total":0
            },
        'MPL' : {
            "Moyen":0.0,
            "Total":0
            },
        'NCY' : {
            "Moyen":0.0,
            "Total":0
            },
        'NAN' : {
            "Moyen":0.0,
            "Total":0
            },
        'NCE' : {
            "Moyen":0.0,
            "Total":0
            },
        'PAR' : {
            "Moyen":0.0,
            "Total":0
            },
        'REN' : {
            "Moyen":0.0,
            "Total":0
            },
        'STG' : {
            "Moyen":0.0,
            "Total":0
            },
        'TLS' : {
            "Moyen":0.0,
            "Total":0
            }
        }
}

i = 0
idi = 0
old = 0
sub = []
for fild in kek:
    if (fild["gpa"] != "n/a"):
        idi += 1
        fild["id"] = idi
        if (fild["gpa"] != old):
            i += 1
            old = fild["gpa"]
        fild["rank"] = i
        fild["gpa"] = float(fild["gpa"])
        if (fild["name"] == "Adrien Zemma"):
            obj["rank Adrien"] = fild["rank"]
        if (fild["name"] == "Hugo Martin"):
            obj["rank Hugo"] = fild["rank"]
        obj["box"][fild["city"]]["Moyen"] += float(fild["gpa"])
        obj["box"][fild["city"]]["Total"] += 1

for _, v in obj["box"].items():
    try:
        v["Moyen"] /= v["Total"]
        v["Moyen"] = round(v["Moyen"], 2)
    except:
        pass

pprint(obj)
print("total rank: ", i)
print("total eleve: ", len(kek))

pew = open("rankSorted.txt","w")
for fild in kek:
    if (fild["gpa"] != 'n/a'):
        line = str(str(fild["rank"]) + ". " + fild["city"] + " " + fild["email"] + ": " + str(fild["gpa"]) + " gpa" + "\n")
        pew.writelines(line)

pew = open("salut.json","w")
pew.write(str(json.dumps(kek, indent=3)))