# I. 549
# Simon Tamás
# 11. osztály
# Kiskunfélegyháza, Kiskunfélegyházi Szent Benedek PG Két Tan. Nyelvű Techn. és Koll.
# Python 3.10.0

import sys
import re

fajlnev = sys.argv[1]
kulcs = sys.argv[2]

alphabet = "AÁBCDEÉFGHIÍJKLMNOÓÖŐPQRSTUÚÜŰVWXYZ"

bemenet = []

with open(fajlnev, encoding="UTF-8") as f:
    bemenet = f.readlines()
    bemenet = [re.sub("[^A-ZÁÉÍÓÖŐÚÜŰ]+", "", sor.upper()) for sor in bemenet]


kimenet = []

sor_index = 0
for sor in bemenet:
    global_index = 0
    kimenet.append([])
    for betu in sor:
        index_betu = alphabet.index(betu)
        index_kulcs = alphabet.index(kulcs[global_index % len(kulcs)])
        index_kodolt = (index_betu + index_kulcs + 1) % len(alphabet)

        kimenet[sor_index] += alphabet[index_kodolt]
        global_index += 1
    kimenet[sor_index] = "".join(kimenet[sor_index]) + "\n"
    sor_index += 1
    

fajlnev_nev = ".".join(fajlnev.split(".")[:-1])
fajlnev_kiterjesztes = fajlnev.split(".")[-1]
kimeneti_fajlnev = fajlnev_nev + "_vc." + fajlnev_kiterjesztes

with open(kimeneti_fajlnev, "w", encoding="UTF8") as f:
    f.writelines(kimenet)