-
Notifications
You must be signed in to change notification settings - Fork 0
/
encrypt-decrypt.py
46 lines (39 loc) · 1.43 KB
/
encrypt-decrypt.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
from art import logo
from a import alphabet
print(logo)
def extend_alphabet(text,shift):
last_letters = {'v','w','x','y','z'}
if any(letter in last_letters for letter in text):
alphabet.extend(alphabet)
def encrypt(text, shift):
encrypted_text = ""
for letters in text:
if letters in alphabet:
position = alphabet.index(letters)+shift
encrypted_text += alphabet[position]
else:
encrypted_text += letters
print(f"The encoded text is {encrypted_text}")
def decrypt(text, shift):
decrypted_text = ""
for letters in text:
if letters in alphabet:
position = alphabet.index(letters)-shift
decrypted_text+=alphabet[position]
else:
decrypted_text+=letters
print(f"The decoded text is {decrypted_text}")
close_program = False
while not close_program:
direction = input("Type 'encode' to encrypt, type 'decode' to decrypt:\n")
text = input("Type your message:\n").lower()
shift = int(input("Type the shift number:\n"))
extend_alphabet(text, shift)
if direction == "encode":
encrypt(text, shift)
elif direction == "decode":
decrypt(text, shift)
restart = input("Type 'yes' to want to go again, type 'no' to want to quit. \n").lower()
if restart == "no":
close_program= True
print("Thanks for using Caesar Cipher!")