Add erro treatment

Author: rjLelis

file-operations/caesar_cipher.py

@@ -17,7 +17,11 @@ def encrypt(original_message, key):
     encrypted_message = ''
 
     if type(key) == str:
-        key = ALPHABET.index(key) + 1
+        try:
+            key = ALPHABET.index(key) + 1
+
+        except ValueError:
+            raise ValueError()
 
     # Encrypts the original message then stores in the return variable
     for letter in remove_accents(original_message.lower()):
@@ -46,7 +50,11 @@ def decrypt(encrypted_message, key):
     translated = ''
 
     if type(key) == str:
-        key = ALPHABET.index(key) + 1
+        try:
+            key = ALPHABET.index(key) + 1
+
+        except ValueError:
+            raise ValueError()
 
     for letter in remove_accents(encrypted_message.lower()):
         if letter in ALPHABET:

file-operations/file_encrypt.py

@@ -33,6 +33,8 @@ def encrypt(file, key):
 
     except FileNotFoundError:
         print(f'The file {file} does not exists')
+    except ValueError:
+        print('The selected key does not exists in the alphabet')
 
 
 
@@ -41,14 +43,16 @@ def decrypt(file, key):
         file_content = get_file_content(file)
 
         if file_content:
-            new_file = store_file_content(caesar_cipher.decrypt(file_content, key), 'decriptado.txt')
+            new_file = store_file_content(caesar_cipher.decrypt(file_content, key), 'decripted.txt')
             print(f'Content stored on {new_file}')
 
         else:
             print(f'The file {file} is empty')
 
     except FileNotFoundError:
         print(f'The file {file} does not exists')
+    except ValueError:
+        print('The selected key does not exists in the alphabet')
 
 
 if __name__ == "__main__":