added features

Author: ppeuchin

caesar_cipher.py

@@ -3,6 +3,7 @@
 from extras import decor
 import time
 from option_validator import options
+from redo import redo
 
 @decor
 def display():
@@ -16,7 +17,11 @@ def display():
 print("Hello there! What can i help you with?")
 time.sleep(1)
 print("\n1. Encryption")
-print("2. Decryption\n")
+print("2. Decryption")
+print("3. Exit\n")
 time.sleep(1)
 
-options()
+options()
+print('\nPress any key to continue')
+input()
+redo()

caesar_decryption.py

@@ -7,6 +7,10 @@ def intro():
     time.sleep(1)
     print("   Welcome To Caesar Cipher Decryptor")
 
+def note():
+    print()
+    print("NOTE: YOU ALWAYS NEED TO HAVE THE SAME KEY USED TO ENCRYPT THE TEXT!")
+    
 def caesar_decryption(ciphertext, key):
     decrypted_str = ""
 

caesar_encryption.py

@@ -7,6 +7,10 @@ def intro():
     time.sleep(1)
     print("   Welcome To Caesar Cipher Encryptor")
 
+def note():
+    print()
+    print("NOTE: ALWAYS REMEMBER TO KEEP THE KEY SECURE! YOU NEED THE SAME KEY TO DECRYPT THE TEXT!")
+
 def caesar_encryption(plaintext, key):
     encrypted_str = ""
 

extras.py

@@ -13,26 +13,11 @@ def load_a():
         sys.stdout.write(i)
         sys.stdout.flush()
         time.sleep(0.3)
-    print()
-
-def load_b():
-    for j in '..............':
-        sys.stdout.write(j)
-        sys.stdout.flush()
-        time.sleep(0.3)
-    print()
-
-def load_c():
-    print('Almost there!')
-    for k in '..............':
-        sys.stdout.write(k)
-        sys.stdout.flush()
-        time.sleep(0.3)
     print('\n')
 
-def load_d():
+def load_b():
     for i in 'Decrypting....':
         sys.stdout.write(i)
         sys.stdout.flush()
         time.sleep(0.3)
-    print()
+    print('\n')

option_validator.py

@@ -1,6 +1,7 @@
 import time
 import os
-from extras import load_a, load_b, load_c, load_d
+import sys
+from extras import load_a, load_b
 import caesar_encryption as ce
 import caesar_decryption as cd
 
@@ -11,6 +12,7 @@ def options():
         time.sleep(1.5)
         os.system('cls')
         ce.intro()
+        ce.note()
         print()
         time.sleep(1)
         plaintext = input("Enter the text: ")
@@ -20,14 +22,13 @@ def options():
         time.sleep(1)
         print("--------------------------")
         load_a()
-        load_b()
-        load_c()
         ce.caesar_encryption(plaintext, key)
 
     elif option == '2':
         time.sleep(1.5)
         os.system('cls')
         cd.intro()
+        cd.note()
         print()
         time.sleep(1)
         ciphertext = input("Enter the encrypted text: ")
@@ -36,11 +37,15 @@ def options():
         print()
         time.sleep(1)
         print("--------------------------")
-        load_d()
         load_b()
-        load_c()
         cd.caesar_decryption(ciphertext, key)
 
+    elif option == '3':
+        time.sleep(1.5)
+        os.system('cls')
+        msg = "Thank you! :)\nSee you next time!"
+        sys.exit(msg)
+
     else:
         time.sleep(1.5)
         print("Invalid Option.")