Morse

MorseCode = {'a': '.-', "а": ".-",
             'b': '-...', "б": "-...",
             'c': '-.-.', "в": ".--",
             'd': '-..', "г": "--.",
             'e': '.', "д": "-..",
             'f': '..-.', "е": ".",
             'g': '--.', "ж": "...-",
             'h': '....', "з": "--..",
             'i': '..', "и": "..",
             'j': '.---', "й": ".---",
             'k': '-.-', "к": "-.-",
             'l': '.-..', "л": ".-..",
             'm': '--', "м": "--",
             'n': '-.', "н": "-.",
             'o': '---', "р": ".-.",
             'p': '.--.', "с": "...",
             'q': '--.-'', "т": "-",
             'r': '.-.', "у": "..-",
             's': '...', "ф": "..-.",
             't': '-', "х": "....",
             'u': '..-', "ц": "-.-.",
             'v': '...-', "ч": "---.",
             'w': '.--', "ш": "----",
             'x': '-..-', "щ": "--.-",
             'y': '-.--', "ъ": ".--.-.",
             'z': '--..', "э": "...-...",
             "ы": "-.--", "ь": "-..-", "я": ".-.-", "о": "---",
             "п": ".--.", "ю": "..--",
             "1": ".----", "2": "..---", "3": "...--", "4": "....-",
             "5": ".....", "6": "-....", "7": "--...",
             "8": "---..", "9": "----.", "0": "-----"}

def decode_from_Morse(code): #Функция декодировки
    d = {}
    for j in MorseCode:
        d[MorseCode.get(j)] = j #Создаётся "перевёрнутый" словарь
    decode = [d.get(j) for j in code.split()]
    need = ''.join(i for i in decode) #Декодированный текст склеивается в одну строку
    return need

def encode_to_Morse(text):
    text = text.lower().split()
    answer = []
    for letter in text:
        answer.append(MorseCode[letter])
    boomer = " ".join(answer)
    return boomer

def main(): #Главная часть кода
    print('Ты будешь "декодировать" или "кодировать"?')
    if input() == "декодировать": #Спрашивается, что выполнить. Далее ссылки на другие функции.
        print(decode_from_Morse(input()))
    else:
        text = input()
        print(encode_to_Morse(input()))

main()