mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-10-06 21:59:30 +00:00
60895366c0
* fix(mypy): type annotations for cipher algorithms * Update mypy workflow to include cipher directory * fix: mypy errors in hill_cipher.py * fix build errors
64 lines
1.7 KiB
Python
64 lines
1.7 KiB
Python
from typing import Literal
|
|
|
|
LETTERS = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
|
|
|
|
|
def translate_message(
|
|
key: str, message: str, mode: Literal["encrypt", "decrypt"]
|
|
) -> str:
|
|
"""
|
|
>>> 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() -> None:
|
|
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()
|