# ALPHA = "abcdefghijklmnopqrstuvwxyz" + "abcdefghijklmnopqrstuvwxyz".upper()
ALPHA = "абвгдеёжзийклмнопрстуфхцчшщъыьэюя" + "абвгдеёжзийклмнопрстуфхцчшщъыьэюя".upper()
SHIFT = 5 # сдвиг

def welcome():
    message = "Добро пожаловать в программу!\n"
    option = "1. Зашифровать\n 2. Расшифровать\n 3. Выйти \n"
    print(message, option)

def output_to_file(string):
    with open("encode.txt", "w") as f:
        f.write(string)

def encode(string, shift):
    string_encode = ""
    for letter in string:
        if ALPHA.count(letter):
            index = ALPHA.index(letter)
            if index > len(ALPHA) - shift:
                string_encode += ALPHA[index - len(ALPHA) + shift]
            else:
                string_encode += ALPHA[index + shift]
        else:
            string_encode += letter
    return string_encode

def decode(string, shift):
    string_decode = ""
    for letter in string:
        if ALPHA.count(letter):
            index = ALPHA.index(letter)
            if index < shift:
                string_decode += ALPHA[index + len(ALPHA) - shift]
            else:
                string_decode += ALPHA[index - shift]
        else:
            string_decode += letter
    return string_decode

def main():
    welcome()
    while True:
        user = input("Введите: ")
        if user == "1":
            user_encode = input("Введите строку для шифрования: ")
            if user_encode:
                user_encode = encode(user_encode, SHIFT)
                output_to_file(user_encode)
                print(user_encode)
            else:
                print("Ошибка ввода : пустая строка")
            print("==============")
        elif user == "2":
            user_decode = input("Введите строку для расшифровки: ")
            if user_decode:
                user_decode = decode(user_decode, SHIFT)
                print(user_decode)
            else:
                print("Ошибка ввода : пустая строка")
            print("==============")
        elif user == "3":
            print("Bye Bye!")
            print("==============")
            break
        else:
            print("Ошибка Ввода! Введите 1 или 2 или 3")
            print("==============")

main()