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

# -*- coding: utf-8 -*-
 
def c_vishener ():
  ABC = ['А','Б','В','Г','Д','Е','Ё','Ж','З','И','Й','К','Л','М','Н','О','П','Р','С','Т','У','Ф','Х','Ц','Ч','Ш','Щ','Ъ','Ы','Ь','Э','Ю','Я']
  print ()
  def switch(n):
    while n<34:
      for i in ABC:
        if i==ABC[0]:
          print(end="|")
        print(i,end="|")
      ABC.append(ABC[0])
      ABC.remove(ABC[0])
      print()
      n+=1
  switch(0)
  print()
  print("Введите текст и ключ для шифрования")
  try:
    with open("key.txt", "w") as file_key:
      m=input("Введите текст: ")
      k=input("Введите ключ: ")
      k*=len(m)//len(k)+1
      file_key.write(k)
    with open("crypt.txt", "w") as file_crypt:
      c=""
      for i,j in enumerate(m):
        gg=(ord(j)+ord(k[i]))
        c+=chr(gg%33+ord('А'))
      print()
      print("Зашифрованный текст: "+str.lower(c))
      file_crypt.write(c)
    v=""
    for i,j in enumerate(c):
      gg=(ord(j)-ord(k[i]))
      v+=chr(gg%33+ord('А'))
    print("Расшифрованный текст: "+str.lower(v)+"\n")
    print("Зашифрованный файл: crypt.txt")
    print("Ключ-файл: key.txt\n")
  except FileNotFoundError: print("[x] Error!"); print("Текст/ключ не определены!\n")
c_vishener()

from itertools import cycle
#from string import ascii_uppercase
 
def wcrypt(word, key, sign):
    # для другого алфавита можно строку из букв алфавита
    alph = "АБВГДЕЁЖЗИКЛМНОПРСТУФХЦЧШЩЪЫЬЭЮЯ"
    n = len(alph)
    gen_key = (alph.index(v) 
            for v in cycle(key.upper()))
    fun = lambda x: alph[(alph.index(x) +
                 sign*next(gen_key))%n]
                 
    res = ''
    for w in word:
        if w in alph:
            res += fun(w)
        elif w.upper() in alph:
            res += fun(w.upper()).lower()
        else:
            res += w
 
    return res
 
 
stri = 'Преступление и наказание'
k = 'Фёдор'
 
# 1 -> шифрование
stri = wcrypt(stri, k, 1)
print(stri)
 
# -1 -> дешифрование
stri = wcrypt(stri, k, -1)
print(stri)