first

2023-08-22 13:29:32 -03:00
commit a3306596e1
2 changed files with 136 additions and 0 deletions
@@ -0,0 +1,76 @@
+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",
+    "à": "a",
+    "é": "e",
+    "ê": "e",
+    "í": "i",
+    "ó": "o",
+    "õ": "o",
+    "ô": "o",
+    "ú": "u",
+    "ç": "c"
+}
+special_chars = {
+    " ": " ",
+    ".": ".",
+    ",": ",",
+    ";": ";",
+    ":": ":",
+    "!": "!",
+    "?": "?",
+    "(": "(",
+    ")": ")"
+}
+
+def cesar(string, key):
+    new_string = ""
+
+    for letter in string:
+        if letter in special_chars:
+            new_string += special_chars[letter]
+            continue
+
+        if letter in special_letters:
+            letter = special_letters[letter]
+
+        new_letter = ord(letter) + 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
+
+        new_string += chr(new_letter)
+
+    return new_string
+
+print(cesar(string, key))