"""Encode and decode Caesar cipher."""


def decode(message: str, shift: int, alphabet="abcdefghijklmnopqrstuvwxyz"):
    """Ayy 10/10."""
    if len(alphabet) != 0:
        shift = shift % len(alphabet)
    alphabet = alphabet
    alphabet_caps = alphabet.upper()
    encoded_word = []
    x = ""
    for encoded_letter in range(len(message)):
        if message[encoded_letter] in alphabet:
            encoded_word.append(alphabet[alphabet.index(message[encoded_letter]) - shift])
            x += encoded_word[-1]
        elif message[encoded_letter] in alphabet_caps:
            encoded_word.append(alphabet_caps[alphabet_caps.index(message[encoded_letter]) - shift])
            x += encoded_word[-1]
        else:
            encoded_word.append(message[encoded_letter])
            x += encoded_word[-1]
    return x


def encode(message: str, shift: int, alphabet="abcdefghijklmnopqrstuvwxyz"):
    """Ayy 10/10."""
    if len(alphabet) != 0:
        shift = shift % len(alphabet)
    alphabet = alphabet * 2
    alphabet_caps = alphabet.upper()
    encoded_word = []
    x = ""
    for encoded_letter in range(len(message)):
        if message[encoded_letter] in alphabet:
            encoded_word.append(alphabet[alphabet.index(message[encoded_letter]) + shift])
            x += encoded_word[-1]
        elif message[encoded_letter] in alphabet_caps:
            encoded_word.append(alphabet_caps[alphabet_caps.index(message[encoded_letter]) + shift])
            x += encoded_word[-1]
        else:
            encoded_word.append(message[encoded_letter])
            x += encoded_word[-1]
    return x


print(decode("afmmp", -1))
print(encode("test 123", -1))  # ifmmp