cesar.py

import string

alfabetoMinusculo = string.ascii_lowercase
alfabetoMaiusculo = string.ascii_uppercase

def encode(message, decode=False):
  result = ''
  for l in message:
    if l in alfabetoMinusculo or l in alfabetoMaiusculo:
      posicao = getLetterPosition(l)
      posicao = ((posicao + 4) if decode else (posicao - 4)) % 26
      result = result + (alfabetoMaiusculo[posicao].upper() if l.isupper() else alfabetoMinusculo[posicao])
    else:
      result = result + l
  return result

def getLetterPosition(letter):
  return alfabetoMinusculo.find(letter) if letter.islower() else alfabetoMaiusculo.find(letter)

msg = 'testando mensagem para ser cifrada'
print(msg)
print(encode(msg))
print(encode(encode(msg), True))