Update JSON format and search

Author: abhijeetpandit7

Day-29-PasswordManager/README.md

@@ -3,4 +3,4 @@
 ### An implementation of Password-Manager
 ### [PasswordManager](https://repl.it/@abhijeetpandit/PasswordManager#main.py)
 
-<img src= 'https://user-images.githubusercontent.com/65078610/106907740-97e84f80-6724-11eb-9128-efced1349621.gif' width="500">
+<img src= 'https://user-images.githubusercontent.com/65078610/107014546-c5370b00-67c1-11eb-81d4-a97f2d9a44da.gif' width="500">

Day-29-PasswordManager/data.json

@@ -0,0 +1,18 @@
+{
+    "Pantaloons": {
+        "email": "youremail@email.com",
+        "password": "UT*l2qoWH7$2Bb6"
+    },
+    "Reliance": {
+        "email": "youremail@email.com",
+        "password": "B(6aIBE9a$0X(N*0J"
+    },
+    "Myntra": {
+        "email": "youremail@email.com",
+        "password": "letsdoshopping"
+    },
+    "Facebook": {
+        "email": "youremail@email.com",
+        "password": "Pl1lf6&HX!9FP"
+    }
+}

Day-29-PasswordManager/main.py

@@ -1,13 +1,73 @@
 from tkinter import *
 import pyperclip
+import json
 from tkinter import messagebox
 from password_gen import generatePassword
 LBLUE = '#eff8ff'
 BLUE = '#cae4db'
 
 # ---------------------------- PASSWORD GENERATOR ------------------------------- #
+def genPassword():
+    password_entry.delete(0,END)
+    new_password = generatePassword()
+    # Copy password to clipboard
+    pyperclip.copy(new_password)
+    password_entry.insert(index=0, string=new_password)
 
 # ---------------------------- SAVE PASSWORD ------------------------------- #
+def addPassword():
+    website = website_entry.get()
+    email = email_entry.get()
+    password = password_entry.get()
+    new_data = {
+        website : {
+            'email' : email,
+            'password' : password
+        }
+    }
+    # Check for empty fields
+    if website=='' or email=='' or password=='':
+        messagebox.showwarning(title='Opps!', message="Please don't leave any field empty")
+    else:
+        # Show confirmation popup
+        is_ok = messagebox.askokcancel(
+            title=f'{website}', 
+            message=f'Are you sure? \nEmail: {email} \nPassword: {password}'
+        )
+        # Add new data to data.json
+        if is_ok:
+            try:
+                with open('data.json','r') as f:
+                    # Read old data
+                    data = json.load(f)
+            except (FileNotFoundError, json.decoder.JSONDecodeError):
+                # If no data found
+                data = new_data
+            else:
+                # Update old data with new data
+                data.update(new_data)
+            finally:
+                with open('data.json','w') as f:
+                    # Saving new data to json
+                    json.dump(data, f, indent=4)
+                website_entry.delete(0,END)
+                password_entry.delete(0,END)
+                website_entry.focus()
+
+# ---------------------------- FIND PASSWORD ------------------------------- #
+def findPassword():
+    website = website_entry.get()
+    try:
+        with open('data.json','r') as f:
+            data = json.load(f)
+            email = data[website]['email']
+            password = data[website]['password']
+            messagebox.askokcancel(
+                title=f'{website}', 
+                message=f'Email: {email} \nPassword: {password}'
+            )
+    except (KeyError, FileNotFoundError):
+        messagebox.showwarning(title='Error', message=f"No details for {website} exists.")
 
 # ---------------------------- UI SETUP ------------------------------- #
 window = Tk()
@@ -31,44 +91,20 @@
 
 # Entry
 website_entry = Entry()
-website_entry.grid(row=2, column=2, padx=10, pady=5, columnspan=2, sticky='EW')
+website_entry.grid(row=2, column=2, padx=10, pady=5, sticky='EW')
 website_entry.focus()
 email_entry = Entry()
 email_entry.grid(row=3, column=2, padx=10, pady=5, columnspan=2, sticky='EW')
 email_entry.insert(index=END, string='youremail@email.com')
 password_entry = Entry()
 password_entry.grid(row=4, column=2, padx=10, pady=5, sticky='EW')
 
-def genPassword():
-    password_entry.delete(0,END)
-    new_password = generatePassword()
-    # Copy password to clipboard
-    pyperclip.copy(new_password)
-    password_entry.insert(index=0, string=new_password)
-
-def addPassword():
-    # Check for empty fields
-    if website_entry.get()=='' or email_entry.get()=='' or password_entry.get()=='':
-        messagebox.showwarning(title='Opps!', message="Please don't leave any field empty")
-    else:
-        # Show confirmation popup
-        is_ok = messagebox.askokcancel(
-            title=f'{website_entry.get()}', 
-            message=f'Are you sure? \nEmail: {email_entry.get()} \nPassword: {password_entry.get()}'
-        )
-        # Add new data to data.txt
-        if is_ok:
-            with open('data.txt','a') as f:
-                # Using print() to write files
-                print(f"{website_entry.get()} | {email_entry.get()} | {password_entry.get()}", file=f)
-            website_entry.delete(0,END)
-            password_entry.delete(0,END)
-            website_entry.focus()
-
 # Button
 gen_password_btn = Button(text='Generate Password', command=genPassword, bg=BLUE)
 gen_password_btn.grid(row=4, column=3)
 add_btn = Button(text='Add', command=addPassword, bg=BLUE)
 add_btn.grid(row=5, column=2, columnspan=2, sticky='EW')
+search_btn = Button(text='Search', command=findPassword, bg=BLUE)
+search_btn.grid(row=2, column=3, columnspan=2, sticky='EW')
 
 window.mainloop()