question_str = "Y"

while question_str == "Y" :

    result_str = "" #Empty string for the result of the encryption or decryption

    message_str = input("Enter Message: ")

    shift_str = int(input("Enter shift amount: "))

    code_str = input("Encode (E) or Decode (D)? ")

    if code_str == "E" :

        message_str = message_str.replace('e', 'zw')
#CREATE SPECIAL ZERO RULE
        middle_int = len(message_str)//2

        message_str = ('hokie' + message_str[:middle_int] + 'hokie' + message_str[middle_int:] + 'hokie')

        for e in message_str :

            char_int = ord(e)

            char_int_changed = ord(e)

            if e.isalpha() :

                if e.isupper() :

                    for x in range(char_int, char_int + shift_str) :

                        char_int_changed += 1

                        if char_int_changed > 90 :

                            char_int_changed = 65

                        result_str = result_str + chr(char_int_changed)

                elif e.islower() :

                    for x in range(char_int, char_int + shift_str) :

                        char_int_changed += 1

                        if char_int_changed > 122 :

                            char_int_changed = 97

                        result_str = result_str + chr(char_int_changed)

            else:

                result_str = result_str + chr(char_int_changed)

        print("Result: ", result_str)

    elif code_str == "D" :

        decrypt_str = ""

        for e in message_str :

            char_int = ord(e)

            char_int_changed = ord(e)

            if e.isalpha() :

                if e.isupper() :

                    for x in range(char_int, char_int - shift_str) :

                        char_int_changed -= 1

                        if char_int_changed < 65 :

                            char_int_changed = 90

                        decrypt_str = decrypt_str + chr(char_int_changed)

                if e.islower() :

                    for x in range(char_int, char_int - shift_str) :

                        char_int_changed -= 1

                        if char_int_changed < 97 :

                            char_int_changed = 122

                        decrypt_str = decrypt_str + chr(char_int_changed)

            else:

                decrypt_str = decrypt_str + chr(char_int_changed)

            print(decrypt_str)


        middle_str = len(decrypt_str)//2

        if middle_str % 2 == 0 :

            middle_str -= 3

        else:

            middle_str -= 2

        result_str = decrypt_str[5:middle_str] + decrypt_str[middle_str+5:len(decrypt_str)-5]

        result_str = result_str.replace('zw','e')

        print("Result: ", result_str)