bug fixes

Author: ppeuchin

extras.py

@@ -1,5 +1,6 @@
 import sys
 import time
+from os import system, name
 
 def decor(func):
     def wrap():
@@ -20,4 +21,12 @@ def load_b():
         sys.stdout.write(i)
         sys.stdout.flush()
         time.sleep(0.3)
-    print('\n')
+    print('\n')
+
+def clear():
+    # Windows
+    if name == 'nt':
+        system('cls')
+    # Mac and Linux
+    else:
+        system('clear')

option_validator.py

@@ -1,7 +1,6 @@
 import time
-import os
 import sys
-from extras import load_a, load_b
+from extras import load_a, load_b,clear
 import caesar_encryption as ce
 import caesar_decryption as cd
 
@@ -10,7 +9,7 @@ def options():
     print()
     if option == '1':
         time.sleep(1.5)
-        os.system('cls')
+        clear()
         ce.intro()
         ce.note()
         print()
@@ -26,7 +25,7 @@ def options():
 
     elif option == '2':
         time.sleep(1.5)
-        os.system('cls')
+        clear()
         cd.intro()
         cd.note()
         print()
@@ -42,7 +41,7 @@ def options():
 
     elif option == '3':
         time.sleep(1.5)
-        os.system('cls')
+        clear()
         msg = "Thank you! :)\nSee you next time!"
         sys.exit(msg)
 

redo.py

@@ -1,9 +1,9 @@
 import time
 from option_validator import options
-import os
+from extras import clear
 
 def redo():
-    os.system('cls')
+    clear()
     print('*******************************')
     print('What would you like to do next?')
     time.sleep(1)