cryptogram.py

def cryptogram(plaintext):
    from random import shuffle
    # create cryptogram cypher. You can't shuffle strings, so lists it is.
    alpha = list('ABCDEFGHIJKLMNOPQRSTUVWXYZ')
    beta = alpha[:]
    shuffle(beta)
    cmap = dict(zip(alpha,beta))
    # Map the received string to a cryptogram. Note we're not cleaning any input.    
    cryptext = [ cmap[c] if c in alpha else c for c in plaintext.upper()]
    return ''.join(cryptext)