Шифр простой замены

import string
word = input('Введите слово для шифрования') # шифруемое слово
codeword = input('введите код слово 9 букв (буквы в слове не должны повторяться: )')  # кодовое слово
alphabet = "абвгдеёжзийклмнопрстуфхцчшщъыьэюя"  # алфавит
new_alphabet = codeword + ''.join([x for x in alphabet if x not in codeword]) # формула подставляет кодовое слово в начало нового алфавита
dct = dict(zip(alphabet, new_alphabet))  # словарь замены символов
new_word = ''.join(map(dct.get, word))  # слово после шифрования
print("Зашифрованное сообщение:", new_word) # вывод результата