mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-24 05:21:09 +00:00
1f8a21d727
* Tighten up psf/black and flake8
* Fix some tests
* Fix some E741
* Fix some E741
* updating DIRECTORY.md
Co-authored-by: github-actions <${GITHUB_ACTOR}@users.noreply.github.com>
79 lines
1.7 KiB
Python
79 lines
1.7 KiB
Python
import random
|
|
import sys
|
|
|
|
LETTERS = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
|
|
|
|
|
def main():
|
|
message = input("Enter message: ")
|
|
key = "LFWOAYUISVKMNXPBDCRJTQEGHZ"
|
|
resp = input("Encrypt/Decrypt [e/d]: ")
|
|
|
|
checkValidKey(key)
|
|
|
|
if resp.lower().startswith("e"):
|
|
mode = "encrypt"
|
|
translated = encryptMessage(key, message)
|
|
elif resp.lower().startswith("d"):
|
|
mode = "decrypt"
|
|
translated = decryptMessage(key, message)
|
|
|
|
print("\n{}ion: \n{}".format(mode.title(), translated))
|
|
|
|
|
|
def checkValidKey(key):
|
|
keyList = list(key)
|
|
lettersList = list(LETTERS)
|
|
keyList.sort()
|
|
lettersList.sort()
|
|
|
|
if keyList != lettersList:
|
|
sys.exit("Error in the key or symbol set.")
|
|
|
|
|
|
def encryptMessage(key, message):
|
|
"""
|
|
>>> encryptMessage('LFWOAYUISVKMNXPBDCRJTQEGHZ', 'Harshil Darji')
|
|
'Ilcrism Olcvs'
|
|
"""
|
|
return translateMessage(key, message, "encrypt")
|
|
|
|
|
|
def decryptMessage(key, message):
|
|
"""
|
|
>>> decryptMessage('LFWOAYUISVKMNXPBDCRJTQEGHZ', 'Ilcrism Olcvs')
|
|
'Harshil Darji'
|
|
"""
|
|
return translateMessage(key, message, "decrypt")
|
|
|
|
|
|
def translateMessage(key, message, mode):
|
|
translated = ""
|
|
charsA = LETTERS
|
|
charsB = key
|
|
|
|
if mode == "decrypt":
|
|
charsA, charsB = charsB, charsA
|
|
|
|
for symbol in message:
|
|
if symbol.upper() in charsA:
|
|
symIndex = charsA.find(symbol.upper())
|
|
if symbol.isupper():
|
|
translated += charsB[symIndex].upper()
|
|
else:
|
|
translated += charsB[symIndex].lower()
|
|
else:
|
|
translated += symbol
|
|
|
|
return translated
|
|
|
|
|
|
def getRandomKey():
|
|
key = list(LETTERS)
|
|
random.shuffle(key)
|
|
return "".join(key)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|