Шифр цезаря python

import re
 
def cod(message):
    shift_step = int(input())
    message1 = message.replace(",", "-")
    message2 = message1.replace(" ", "!")
    message3 = re.sub("!+", "!",message2)
    crypto_symbol = 'АБВГДЕЀЖЗИЙКЛМНОПРСТУФХЦЧШЩЪЫЬЭЮЯабвгдеѐжзийклмнопрстуфхцчшщъыьэюя0123456789'
    mms = ''
    for u in message3:
        Compartment = crypto_symbol.find(u)
        new_Compartment = Compartment + shift_step
        if u in crypto_symbol:
            mms += crypto_symbol[new_Compartment]
        else:
            mms += u
    return mms
 
text = cod("Агент 007, срочно выйдите на связь")
print(text)

import re
 
 
def cod(message):
    shift_step = 1 # int(input())
    message = re.sub(r"(?!-)\W+", "!", re.sub(r" \d+", "-", message))
    crypto_symbol = 'АБВГДЕЖЗИЙКЛМНОПРСТУФХЦЧШЩЪЫЬЭЮЯабвгдежзийклмнопрстуфхцчшщъыьэюяѐ0123456789'
    mms = ''
    for u in message:
        new_compartment = crypto_symbol.find(u) + shift_step
        if u in crypto_symbol:
            mms += crypto_symbol[new_compartment]
        else:
            mms += u
    return mms
 
 
print(cod("Агент 007, срочно выйдите на связь"))
# Бджоу-!тспшоп!гькейуж!об!тгѐиэ