-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.py
More file actions
116 lines (85 loc) · 3.06 KB
/
main.py
File metadata and controls
116 lines (85 loc) · 3.06 KB
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
from tkinter import *
from tkinter import messagebox
import base64
FONT = ("Arial", 20, "bold")
window = Tk()
window.title("Secret Notes")
window.minsize(width=500, height=500)
window.config(padx=30, pady=30)
# * Function to save and encrypt notes
def save_and_encrypt_notes():
title = title_entry.get()
message = input_text.get("1.0", END)
master_secret = master_secret_input.get()
if len(title) == 0 or len(message) == 0 or len(master_secret) == 0:
messagebox.showerror(title="Error", message="Please fill in all fields.")
else:
message_encrypted = encode(master_secret, message)
try:
with open("mysecret.txt", "a") as file:
file.write(f'\n{title}\n{message_encrypted}')
except FileNotFoundError:
with open("mysecret.txt", "w") as file:
file.write(f'\n{title}\n{message_encrypted}')
finally:
title_entry.delete(0, END)
master_secret_input.delete(0, END)
input_text.delete("1.0", END)
# * Function to encode and decode messages
def encode(key, clear):
enc = []
for i in range(len(clear)):
key_c = key[i % len(key)]
enc_c = chr((ord(clear[i]) + ord(key_c)) % 256)
enc.append(enc_c)
return base64.urlsafe_b64encode("".join(enc).encode()).decode()
def decode(key, enc):
dec = []
enc = base64.urlsafe_b64decode(enc).decode()
for i in range(len(enc)):
key_c = key[i % len(key)]
dec_c = chr((256 + ord(enc[i]) - ord(key_c)) % 256)
dec.append(dec_c)
return "".join(dec)
# * Function to decrypt notes
def decrypt_notes():
message_encrypted = input_text.get("1.0", END)
master_secret = master_secret_input.get()
if len(message_encrypted) == 0 or len(master_secret) == 0:
messagebox.showerror(title="Error", message="Please fill in all fields.")
else:
try:
decrypted_message = decode(master_secret, message_encrypted)
input_text.delete("1.0", END)
input_text.insert("1.0", decrypted_message)
except:
messagebox.showerror(title="Error", message="Decryption failed. Check your master secret.")
# * photo
photo = PhotoImage(file="top-secret.png")
photo_label = Label(image=photo)
photo_label.pack()
# * Title Label
title_info_label = Label(text="Enter your title:", font=FONT)
title_info_label.pack()
# * Title Input
title_entry = Entry(width=30)
title_entry.pack()
# * Note Label
input_info_label = Label(text="Enter your secret:")
input_info_label.pack()
# * Note Input
input_text = Text(height=10, width=40)
input_text.pack()
# * Master Secret Label
master_secret_label = Label(text="Enter your master secret:", font=FONT)
master_secret_label.pack()
# * Master Secret Input
master_secret_input = Entry(width=30)
master_secret_input.pack()
# * Save Button
save_button = Button(text="Save & Encrypt", width=10, command=save_and_encrypt_notes)
save_button.pack()
# * Decrypt Button
decrypt_button = Button(text="Decrypt", width=10, command=decrypt_notes)
decrypt_button.pack()
window.mainloop()