17_Cryptography.py

""" Diccionarios """

KEYS = {
    'a': 'w',
    'b': 'E',
    'c': 'x',
    'd': '1',
    'e': 'a',
    'f': 't',
    'g': '0',
    'h': 'C',
    'i': 'b',
    'j': '!',
    'k': 'z',
    'l': '8',
    'm': 'M',
    'n': 'I',
    'o': 'd',
    'p': '.',
    'q': 'U',
    'r': 'Y',
    's': 'i',
    't': '3',
    'u': ',',
    'v': 'J',
    'w': 'N',
    'x': 'f',
    'y': 'm',
    'z': 'W',
    'A': 'G',
    'B': 'S',
    'C': 'j',
    'D': 'n',
    'E': 's',
    'F': 'Q',
    'G': 'o',
    'H': 'e',
    'I': 'u',
    'J': 'g',
    'K': '2',
    'L': '9',
    'M': 'A',
    'N': '5',
    'O': '4',
    'P': '?',
    'Q': 'c',
    'R': 'r',
    'S': 'O',
    'T': 'P',
    'U': 'h',
    'V': '6',
    'W': 'q',
    'X': 'H',
    'Y': 'R',
    'Z': 'l',
    '0': 'k',
    '1': '7',
    '2': 'X',
    '3': 'L',
    '4': 'p',
    '5': 'v',
    '6': 'T',
    '7': 'V',
    '8': 'y',
    '9': 'K',
    '.': 'Z',
    ',': 'D',
    '?': 'F',
    '!': 'B',
    ' ': '&',
}

def cipher_v2(message):
    cipher_message = ''

    for letter in message:
        cipher_message += KEYS[letter]
    
    return cipher_message

def decipher_v2(message):
    decipher_message = ''

    for letter in message:
        for key, value in KEYS.items():
            if value == letter:
                decipher_message += key

    return decipher_message

def cipher_v1(message):
    words = message.split(' ')
    cipher_message = []

    for word in words:
        cipher_word = ''
        for letter in word:
            cipher_word += KEYS[letter]
        
        cipher_message.append(cipher_word)
    
    return ' '.join(cipher_message)

def decipher_v1(message):
    words = message.split(' ')
    decipher_message = []

    for word in words:
        decipher_word = ''
        for letter in word:
            for key, value in KEYS.items():
                if value == letter:
                    decipher_word += key
        
        decipher_message.append(decipher_word)
    
    return ' '.join(decipher_message)

def run():
    command = None
    while command != 's':
        command = str(input('''--- * --- * --- * --- * --- * --- * --- * ---
    Bienvenido a criptografía. ¿Qué deseas hacer?

    [c]ifrar mensaje
    [d]ecifrar mensaje
    [s]alir
'''))

        if command == 'c':
            message = str(input('Escribe tu mensaje: '))
            cipher_message = cipher_v2(message)
            print(cipher_message)
        elif command == 'd':
            message = str(input('Escribe tu mensaje cifrado: '))
            decipher_message = decipher_v2(message)
            print(decipher_message)
        elif command == 's':
            pass
        else:
            print('¡Comando no encontrado!')


if __name__ == '__main__':
    print('M E N S A J E S  C I F R A D O S')
    run()