# 1. feladat
vonatok = []
with open('vonat.txt') as be:
    for sor in be.readlines():
        v_id, a_id, o, p, i = sor.split()
        vonatok.append([int(v_id), int(a_id), int(o), int(p), i])


# 2. feladat
v, a = set(), set()
for v_id, a_id, o, p, i in vonatok:
    v.add(v_id)
    a.add(a_id)
print(f"2. feladat\nAz állomások száma: {len(a)}\n"
      f"A vonatok száma: {len(v)}")

# 3. feladat
legido = 0
legallomas = None
legvonat = None
for vonat in v:
    elozo= 0
    indul, erkezik =  0, 0
    maxallas = 0
    maxvonat, maxallomas = None, None
    for v_id, a_id, o, p, i in vonatok:
        if v_id == vonat :
            if i == 'I' and a_id > 0:
                indul = o * 60 + p
                allas = indul - erkezik
                if allas > maxallas:
                    maxallas = allas
                    maxvonat = vonat
                    maxallomas = a_id
            else:
                erkezik = o * 60 + p
    if maxallas > legido:
        legido = maxallas
        legvonat = maxvonat
        legallomas = maxallomas
print(f"3. feladat\nA(z) {legvonat}. vonat a(z) {legallomas}. állomáson "
      f"{legido} percet állt.")

# 4. feladat
vonat = int(input("Adja meg egy vonat azonosítóját! "))
idopont = input("Adjon meg egy időpontot (óra perc)! ").split()
ora = int(idopont[0])
perc = int(idopont[1])
ut = 2 * 60 + 22

# 5. feladat
for v_id, a_id, o, p, i in vonatok:
    if v_id == vonat:
        if a_id == 0: indulo = o * 60 + p
        else: erkezo = o * 60 + p
menetido = erkezo - indulo
if menetido > ut:
    print(f"5. feladat\nA {vonat}. vonat útja {menetido - ut} perccel hosszabb volt az előírtnál.")
elif menetido < ut:
    print(f"5. feladat\nA {vonat}. vonat útja {ut - menetido} perccel rövidebb volt az előírtnál.")
else:
    print(f"5. feladat\nA {vonat}. vonat útja pontosan az előírt ideig tartott.")

# 6. feladat
with open('halad' + str(vonat) + '.txt', 'w') as ki:
    for v_id, a_id, o, p, i in vonatok:
        if v_id == vonat and i == 'E':
            print(f"{a_id}. allomas: {o}:{p}", file=ki)