tancok = []
try:
    f = open("tancrend.txt")
    i = 0
    egy_tanc = []
    for sor in f:
        i += 1
        egy_tanc.append(sor.strip())
        if i % 3 == 0:
            tancok.append(egy_tanc)
            egy_tanc = []
    f.close()
except FileNotFoundError:
    print("Nincs ilyen fájl")
    exit(-1)
# print(tancok)
print("2. feladat:")
print("Első tánc:", tancok[0][0], "\nUtolsó tánc:", tancok[-1][0])

print("3. feladat")
samba_db = sum(1 for x in tancok if x[0] == "samba")
print("Összesen", samba_db, "pár mutatott be samba-t.")

print("4. feladat")
vilma_db = sum(1 for x in tancok if x[1] == "Vilma")
print("Összesen", vilma_db, "táncban szerepelt Vilma.")

tanc = input("Adja meg a tánc nevét: ")
vilma_parja = [x[2] for x in tancok if x[1] == "Vilma" and x[0] == tanc]
if len(vilma_parja):
    print("A", tanc, "bemutatóján Vilma párja", *vilma_parja, "volt.")
else:
    print("Vilma nem táncolt {}-t.".format(tanc))

print("6. feladat: fájlba írása a szereplőknek")
ki = open("szereplok.txt", "w")
lanyok, fiuk = {}, {}
for tanc, lany, fiu in tancok:
    lanyok[lany] = 0
    fiuk[fiu] = 0
f, l = 0, 0
print("Lányok: ", end="",  file=ki)
for kulcs in lanyok:
    if l < len(lanyok)-1:
        print(kulcs, end=", ", file=ki)
    else:
        print(kulcs, end="", file=ki)
    l += 1
print("\nFiúk: ", end="", file=ki)
for kulcs in fiuk:
    if f < len(fiuk)-1:
        print(kulcs, end=", ", file=ki)
    else:
        print(kulcs, end="", file=ki)
    f += 1
ki.close()
for tanc, lany, fiu in tancok:
    lanyok[lany] += 1
    fiuk[fiu] += 1

print("7. feladat:")
max_lany, max_fiu = max(lanyok.values()), max(lanyok.values())
print("A legtöbbet szereplő fiúk:")
for kulcs, ertek in fiuk.items():
    if ertek == max_fiu:
        print(kulcs)
print("A legtöbbet szereplő lányok:")
for kulcs, ertek in lanyok.items():
    if ertek == max_lany:
        print(kulcs)