diff --git a/ciphers/caesar_cipher.py b/ciphers/caesar_cipher.py index 4560db03b..6cd35e73d 100644 --- a/ciphers/caesar_cipher.py +++ b/ciphers/caesar_cipher.py @@ -1,3 +1,4 @@ +import sys def encrypt(strng, key): encrypted = '' for x in strng: @@ -20,50 +21,48 @@ def decrypt(strng, key): def brute_force(strng): key = 1 decrypted = '' - while key != 96: + while key <= 94: for x in strng: indx = (ord(x) - key) % 256 if indx < 32: indx = indx + 95 decrypted = decrypted + chr(indx) - print(decrypted) + print("Key: {}\t| Message: {}".format(key, decrypted)) decrypted = '' key += 1 return None def main(): - print("**Menu**") + print('-' * 10 + "\n**Menu**\n" + '-' * 10) print("1.Encrpyt") print("2.Decrypt") print("3.BruteForce") print("4.Quit") while True: - choice = input("what would you like to do") + choice = input("What would you like to do?: ") if choice not in ['1', '2', '3', '4']: print ("Invalid choice") elif choice == '1': - strng = input("Please enter the string to be ecrypted:") + strng = input("Please enter the string to be ecrypted: ") while True: - key = int(input("Please enter off-set between 1-94")) - if key > 0 and key <= 94: + key = int(input("Please enter off-set between 1-94: ")) + if key in range(1, 95): print (encrypt(strng, key)) main() elif choice == '2': - strng = input("Please enter the string to be decrypted:") + strng = input("Please enter the string to be decrypted: ") while True: - key = int(input("Please enter off-set between 1-94")) + key = int(input("Please enter off-set between 1-94: ")) if key > 0 and key <= 94: print(decrypt(strng, key)) main() elif choice == '3': - strng = input("Please enter the string to be decrypted:") + strng = input("Please enter the string to be decrypted: ") brute_force(strng) main() elif choice == '4': - print ("GoodBye.") - break + print ("Goodbye.") + sys.exit() main() - -