finalizando trab 1

trabalho1/cesar.py

@@ -1,19 +1,5 @@
 import sys
 
-if len(sys.argv) != 4:
-    print("Usage: python3 cesar.py -c/-d [key] [string]")
-    sys.exit(1)
-
-encrypt = True
-if sys.argv[1] == "-d":
-    encrypt = False
-
-key = int(sys.argv[2])
-string = sys.argv[3]
-
-if not encrypt:
-    key *= -1
-
 special_letters = {
     "ã": "a",
     "á": "a",
@@ -52,29 +38,47 @@ def cesar(string, key):
         if letter in special_letters:
             letter = special_letters[letter]
 
+        if letter not in caracteres:
+            continue
+
         index = caracteres.index(letter)
         new_index = (index + key) % len(caracteres)
         new_letter = caracteres[new_index]
 
-        # # [A-Z, a-z, 0-9]
-        # if ord('0') <= ord(letter) <= ord('9'):
-        #     if new_letter > ord('9'):
-        #         new_letter += 7 # 9 + 1 = A
-        #     elif new_letter < ord('0'):
-        #         new_letter += 75 # 0 - 1 = z
-        # elif letter.isupper():
-        #     if new_letter > ord('Z'):
-        #         new_letter += 6 # Z + 1 = a
-        #     elif new_letter < ord('A'):
-        #         new_letter -= 7 # A - 1 = 9
-        # else:
-        #     if new_letter > ord('z'):
-        #         new_letter -= 75 # z + 1 = 0
-        #     elif new_letter < ord('a'):
-        #         new_letter -= 6 # a - 1 = Z
-
         new_string += new_letter
 
     return new_string
 
-print(cesar(string, key))
+if __name__ == "__main__":
+    if len(sys.argv) != 4:
+        print("Usage: python3 cesar.py -c/-d [key] [string]")
+        sys.exit(1)
+
+    encrypt = True
+    if sys.argv[1] == "-d":
+        encrypt = False
+
+    key = int(sys.argv[2])
+    string = sys.argv[3]
+
+    if not encrypt:
+        key *= -1
+
+    print(cesar(string, key))
+
+# # [A-Z, a-z, 0-9]
+# if ord('0') <= ord(letter) <= ord('9'):
+#     if new_letter > ord('9'):
+#         new_letter += 7 # 9 + 1 = A
+#     elif new_letter < ord('0'):
+#         new_letter += 75 # 0 - 1 = z
+# elif letter.isupper():
+#     if new_letter > ord('Z'):
+#         new_letter += 6 # Z + 1 = a
+#     elif new_letter < ord('A'):
+#         new_letter -= 7 # A - 1 = 9
+# else:
+#     if new_letter > ord('z'):
+#         new_letter -= 75 # z + 1 = 0
+#     elif new_letter < ord('a'):
+#         new_letter -= 6 # a - 1 = Z