81 lines
1.6 KiB
Python
81 lines
1.6 KiB
Python
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 = {
|
|
" ": " ",
|
|
".": ".",
|
|
",": ",",
|
|
";": ";",
|
|
":": ":",
|
|
"!": "!",
|
|
"?": "?",
|
|
"(": "(",
|
|
")": ")"
|
|
}
|
|
|
|
caracteres = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789'
|
|
|
|
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]
|
|
|
|
index = caracteres.index(letter)
|
|
new_index = (index + key) % len(caracteres)
|
|
new_letter = caracteres[new_index]
|
|
|
|
new_string += new_letter
|
|
|
|
return new_string
|
|
|
|
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 |