Кодирование и декодирование методом Цезаря со смещением букв на k позиций текстов, записанных латинскими буквами

lat_alf = 'abcdefghijklmnopqrstuvwxyz'
#==============================================================================
def caesar_encrypt(text, alf, shift ):
    def get_new_lower_let(x):
        return alf[ ( alf.index(x)+shift ) % len(alf) ]
    #==========================================================================
    def get_new_let(x):
        if x.isupper():
            return get_new_lower_let( x.lower() ).upper()
        return get_new_lower_let(x)
    # ==========================================================================
    text = [ x for x in text if x.isalpha() ]
    return ''.join( list( map( get_new_let, text ) ) )
#==============================================================================
def caesar_decrypt(text, alf, shift ):
    return caesar_encrypt( text, alf, -shift )
#==============================================================================
text   = input('Введите сообщение для шифрования: ')
sh     = int( input('Введите сдвиг: ') )
c_code = caesar_encrypt( text, lat_alf, sh )
print(f'Исходное сообщение: {text}')
print(f'Сдвиг: {sh}')
print(f'Зашифрованное сообщение: { c_code }')
print(f'Расшифрованное сообщение: { caesar_decrypt( c_code, lat_alf, sh ) }' )