Fourth project

Author: Parq254

.idea/workspace.xml

@@ -0,0 +1 @@
+ju2kiai0NNxD0ttnpGEB7inKmHUE3E4849Zdnmjnldk=

key.key

@@ -1,2 +1 @@
-facebook|chris
-instagram|ccc
+joe|gAAAAABmYFBwJj1vnPOVvsu2Ea2N92HIqohA57PnEijvDXw7pbaB-hnB_3rm_MalIf56xVrrpATUAvyw4nD-U5rQKmJW4I7jAQ==

password.txt

@@ -1,26 +1,39 @@
+from cryptography.fernet import Fernet
+
+'''
+def write_key():
+    key = Fernet.generate_key()
+    with open('key.key', 'wb') as key_file:
+        key_file.write(key)
+'''
+
+
+def load_key():
+    file = open('key.key', 'rb')
+    key = file.read()
+    file.close()
+    return key
+
+
 master_pwd = input('What is the master password? ')
+key = load_key() + master_pwd.encode()
+fer = Fernet(key)
 
 
 def view():
     with open('password.txt', 'r') as f:
         for line in f.readlines():
             data = line.rstrip()
             user, passw = data.split('|')
-            print('user:', user, ',password:', passw)
-
-
-view()
+            print('user:', user, '| password:', fer.decrypt(passw.encode()).decode())
 
 
 def add():
     name = input('Account name: ')
     pwd = input('Password: ')
 
     with open('password.txt', 'a') as f:
-        f.write(name + '|' + pwd + '\n')
-
-
-add()
+        f.write(name + '|' + fer.encrypt(pwd.encode()).decode() + '\n')
 
 
 while True: