cipher.py

   1 import string
   2
   3
   4 def caesar_cipher_letter(letter, shift):
   5     if letter in string.ascii_letters:
   6         if letter in string.ascii_lowercase:
   7             return chr((ord(letter) - ord('a') + shift) % 26 + ord('a'))
   8         else:
   9             new_letter = letter.lower()
  10             yolo = chr((ord(new_letter) - ord('a') + shift) % 26 + ord('a'))
  11             return yolo.upper()
  12     else:
  13         return letter
  14
  15 def caesar_decipher_letter(letter, shift):
  16     return caesar_cipher_letter(letter, -shift)
  17
  18 def caesar_cipher_message(message, shift):
  19     big_cipher = [caesar_cipher_letter(l, shift) for l in message]
  20     return ''.join(big_cipher)
  21
  22 def caesar_decipher_message(message, shift):
  23     return caesar_cipher_message(message, -shift)