Caesar Crypto

caesar.py

import string

def caesar(plaintext, shift):
    alphabet_lower = string.ascii_lowercase
    alphabet_upper = string.ascii_uppercase
    alphabet = alphabet_lower + alphabet_upper
    shifted_alphabet = alphabet_lower[shift:] + alphabet_lower[:shift] + alphabet_upper[shift:] + alphabet_upper[:shift]
    table = str.maketrans(alphabet, shifted_alphabet)
    return plaintext.translate(table)