import sys, csv, send2trash, json

nazivFajla = sys.argv[1]
file = open(nazivFajla, 'r', encoding='utf-8')
reader = csv.DictReader(file)

spisak = []
temp = 0
ocene = []
broj = 0

for row in reader:
    if (row['IMDb'] != ''):
        ocene.append(float(row['IMDb']))
    if (temp == 0):
        if (int(row['Netflix'])==1):
            spisak.append("Netflix")
        if (int(row['Hulu'])==1):
            spisak.append("Hulu")
        if (int(row['Prime Video'])==1):
            spisak.append("Prime Video")
        if (int(row['Disney+'])==1):
            spisak.append("Disney+")
        temp = 1

file.close()
send2trash.send2trash(nazivFajla)

if (len(ocene) == 0):
    sys.exit()


ocene.sort()
srednja = sum(ocene)/len(ocene)

medijana = 0
if (len(ocene)%2 == 0):
    medijana = (ocene[len(ocene)//2 - 1] + ocene[len(ocene)//2])/2
else:
    medijana = ocene[len(ocene)//2]

povratno = {
    "spisak": spisak,
    "srednja": srednja,
    "medijana": medijana
}

jsonFile = open(nazivFajla.split(".")[0]+".json",'w')
jsonObjekat = json.dumps(povratno,indent=4)

jsonFile.write(jsonObjekat)
jsonFile.close()