Простая программка на питон с шифрованием, пожалуйста подкорректируйте

from functions_v2 import encryption, decryption, reading, writing
 
Key = int(input("Введите ключ шифрования: "))
 
operation = str(input("Введите 'ш' для шифрования, 'р' для расшифрования: "))
 
if operation == 'ш':
  PlainText = reading()
  CipherText = encryption(Key, PlainText) 
  print("\n[+] Шифрование завершено\n") 
  writing(CipherText, operation)
elif operation == 'р' : # если необходимо расшифровать текст
  CipherText = reading()
  PlainText = decryption(Key, CipherText)
  print("\n[+] Расшифрование завершено\n") 
  writing(PlainText, operation)
else:
  print("[-]Выбрана неверная операция")# если введена неверная операция

def reading():
 source = input("Источник данных: \n 'к' - консоль \t 'ф' - файл \n") 
 if source == 'к':
   data = input("Введите текст: ").strip()
   return data
 elif source == 'ф':
   filename = input("Название файла с данными в формате '*.txt': \n")
   with open(filename, 'r') as f:
    data = f.read()
   data = data.encode("cp1251").decode("cp1251")
   return data
 else:
  print("[-] Выбрана неверная операция")
 
def writing(data, operation):
  op = operationtype(operation)
  exit = input("Вид вывода данных: \n 'к' - консоль \t 'ф' - файл: \n")
  if exit == 'к':
    print("Результат: ", data)
  elif exit == 'ф':
    string = f"Файл для вывода результата {op} в формате 'имя.txt': \n" 
    filename = input(string)
    with open(filename, 'w') as f: 
      for simbol in data:
       try:
         f.write(i.encode("cp1251").decode("cp1251"))
       except:
         print(i)
         pass
    print("\n[+] Результат %s записан в файл %s"%(operationtype(operation), filename))
  else:
    print("[-] Выбрана неверная операция")
 
def operationtype(operation): 
  sh = "шифрования"
  rsh = "расшифрования" 
  if operation == "ш": 
     return sh
  elif operation == "р":
     return rsh
 
def encryption(Key, PlainText):
 CipherText = ''
 for p in PlainText:
  code = ord(p)+Key
  CipherText += chr(code)
 return CipherText
 
 
def decryption(Key, CipherText):
 PlainText = ''
 for p in CipherText:
  code = ord(p)-Key
  PlainText += chr(code)
 return PlainText;