mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-12-18 01:00:15 +00:00
2ec3750885
* create monoalphabetic cipher * update file * update file * update file * update file * update file * update after testing flake8 on this code * update file * update file * update file * update file * update file * update file
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
LETTERS = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
|
|
|
|
|
def translate_message(key, message, mode):
|
|
"""
|
|
>>> translate_message("QWERTYUIOPASDFGHJKLZXCVBNM","Hello World","encrypt")
|
|
'Pcssi Bidsm'
|
|
"""
|
|
chars_a = LETTERS if mode == "decrypt" else key
|
|
chars_b = key if mode == "decrypt" else LETTERS
|
|
translated = ""
|
|
# loop through each symbol in the message
|
|
for symbol in message:
|
|
if symbol.upper() in chars_a:
|
|
# encrypt/decrypt the symbol
|
|
sym_index = chars_a.find(symbol.upper())
|
|
if symbol.isupper():
|
|
translated += chars_b[sym_index].upper()
|
|
else:
|
|
translated += chars_b[sym_index].lower()
|
|
else:
|
|
# symbol is not in LETTERS, just add it
|
|
translated += symbol
|
|
return translated
|
|
|
|
|
|
def encrypt_message(key: str, message: str) -> str:
|
|
"""
|
|
>>> encrypt_message("QWERTYUIOPASDFGHJKLZXCVBNM", "Hello World")
|
|
'Pcssi Bidsm'
|
|
"""
|
|
return translate_message(key, message, "encrypt")
|
|
|
|
|
|
def decrypt_message(key: str, message: str) -> str:
|
|
"""
|
|
>>> decrypt_message("QWERTYUIOPASDFGHJKLZXCVBNM", "Hello World")
|
|
'Itssg Vgksr'
|
|
"""
|
|
return translate_message(key, message, "decrypt")
|
|
|
|
|
|
def main():
|
|
message = "Hello World"
|
|
key = "QWERTYUIOPASDFGHJKLZXCVBNM"
|
|
mode = "decrypt" # set to 'encrypt' or 'decrypt'
|
|
|
|
if mode == "encrypt":
|
|
translated = encrypt_message(key, message)
|
|
elif mode == "decrypt":
|
|
translated = decrypt_message(key, message)
|
|
print(f"Using the key {key}, the {mode}ed message is: {translated}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import doctest
|
|
|
|
doctest.testmod()
|
|
main()
|