1
1
from Modules .Colours import *
2
2
from cryptography .fernet import Fernet
3
+ import sys
3
4
4
5
def password_function ():
5
6
6
- file = open ('../temp/key.txt ' ,'rb' )
7
+ file = open ('../temp/key.temp ' ,'rb' )
7
8
key = file .read ()
8
9
file .close ()
10
+ fernet = Fernet (key )
9
11
10
12
try :
11
- file = open ('../temp/password.txt' , 'r' )
12
- password = file .read ()
13
+ with open ("../temp/password.temp" , 'rb' ) as f :
14
+ password = f .read ()
15
+ decrypted = fernet .decrypt (password )
16
+ decrypted = decrypted .decode ()
13
17
while (True ):
14
18
entered_password = input ("Enter your password \n > " )
15
- if (password == entered_password ):
19
+ if (decrypted == entered_password ):
16
20
green ("Logged in successfully" )
17
21
grey ("Press any key to continue..." )
18
22
input ()
@@ -26,11 +30,12 @@ def password_function():
26
30
red ("DO NOT FORGET THIS PASSWORD. YOU CANNOT CHANGE IT LATER." )
27
31
28
32
password = input ("Enter your password \n > " )
29
- fernet = Fernet ( key )
33
+ password = password . encode ( )
30
34
encrypted = fernet .encrypt (password )
31
35
32
- file = open ('../temp/password.txt' , 'w' )
33
- file .write (encrypted )
36
+ with open ("../temp/password.temp" , 'wb' ) as f :
37
+ f .write (encrypted )
38
+
34
39
green ("Your password has been saved successfully" )
35
40
cyan ("Relaunch the application to continue." )
36
41
grey ("Press enter to exit..." )
0 commit comments