# 1. feladat
nyilt_szoveg = input("1. feladat:\nAdjon meg egy szöveget: ").upper()
betuk = {'Á': 'A', 'É': 'E', 'Í': 'I', 'Ó': 'O', 'Ö': 'O', 'Ő': 'O', 'Ü': 'U', 'Ű': 'U', 'Ú': 'U'}

kodolt_szoveg = None
for k, v in betuk.items():
    kodolt_szoveg = nyilt_szoveg.replace(k, v)
    nyilt_szoveg = kodolt_szoveg
# for c in kodolt_szoveg:
#     if c not in ('A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O',
#                  'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z'):
#         kodolt_szoveg = kodolt_szoveg.replace(c, '')
for c in kodolt_szoveg:
    if ord(c) not in range(ord('A'), ord('Z') + 1):
        kodolt_szoveg = kodolt_szoveg.replace(c, '')

# 2 - 3. feladat
print("2, 3. feladat\nAz átalakított szöveg: ", kodolt_szoveg)

# 4. feladat
kulcsszo = input("4. feladat:\nAdjon meg egy legfeljebb ötbetűs kulcsszót: ").upper()

# 5. feladat
print("5. feladat")
hosszu_kulcs = ''
for i in range(len(kodolt_szoveg) // len(kulcsszo)):
    hosszu_kulcs += kulcsszo
hosszu_kulcs += kulcsszo[:len(kodolt_szoveg) % len(kulcsszo)]
print(hosszu_kulcs)

# 6. feladat
print("6. feladat: kódolás")

f = open("Vtabla.dat")
kodtabla = [x.strip() for x in f.readlines()]
f.close()

kod = ''
for i in range(len(hosszu_kulcs)):
    o = [kodtabla[j][0] for j in range(len(kodtabla))]
    oszlop = o.index(kodolt_szoveg[i])
    sor = kodtabla[0].index(hosszu_kulcs[i])
    kod += kodtabla[sor][oszlop]

# 7. feladat
print("7. feladat:\n", kod, sep='')
ki = open("kodolt.dat", "w")
print(kod, file=ki)
ki.close()