Шифр Виженера

def encrypt_ceasar(word, key):
    result = ''
    for lett, k in zip(word, key):
        shift = dct[k.lower()]
        nev_lett = 65 + (ord(lett) - 65 + shift) % 26
        result += chr(nev_lett)
    return result
 
 
def decrypt_ceasar(word, key):
    result = ''
    for lett, k in zip(word, key):
        shift = dct[k.lower()]
        nev_lett = 65 + (ord(lett) - 65 - shift) % 26
        result += chr(nev_lett)
    return result
 
 
def adeq(word, key):
    size_word = len(word)
    while len(key) < size_word:
        key += key
    return key
 
 
dct = {}
start, end = ord('a'), ord('z') + 1
for i in range(start, end):
    dct[chr(i)] = i - start
 
print('Что будем делать?(0 - зашифровывать;1 - расшифровывать)')
act = int(input())
word = input('Введите слово:  ')
key = input('ключ:  ')
nev_key = adeq(word, key)
 
if act == 0:
    print('Зашифрованное слово:', encrypt_ceasar(word, nev_key))
else:
    print('Расшифрованное слово:', decrypt_ceasar(word, nev_key))

from itertools import cycle
from string import ascii_uppercase
 
def w_crypt(word, key, sign=1):
    # для другого алфавита можно строку из букв алфавита
    alph = ascii_uppercase
    n = len(alph)
    
    gen_key = (alph.index(v) for v in cycle(key))
    res = ''
    
    for w in word:
        low = w.islower()
        
        if w.upper() in alph:
            delta = sign*next(gen_key)
            tmp = alph[(alph.index(w.upper()) + delta) % n]
            if low:
                tmp = tmp.lower()
            res += tmp
        else:
            res += w
 
    return res
 
 
def w_decrypt(word, key):
    return w_crypt(word, key, -1)
 
 
stri = 'ATTACK @45im;#DAWN'
k = 'LEMON'
 
# 1 -> шифрование
stri = w_crypt(stri, k)
print(stri)
 
# -1 -> дешифрование
stri = w_decrypt(stri, k)
print(stri)