update password generator to now include exception and error handling

Author: paulipotter

Password-Generator-v2/main.py

@@ -1,3 +1,4 @@
+import json
 from tkinter import *
 from tkinter import messagebox
 from random import choice, shuffle, randint
@@ -23,20 +24,35 @@ def generate_password():
 
 
 # ---------------------------- SAVE PASSWORD ------------------------------- #
-def add_information():
+def save():
     website = website_input.get()
     username = username_input.get()
     password = password_input.get()
 
     # Validate Entries
     if validate_entries(website, username, password):
-        account_data = f"{website} | {username} | {password}"
-        print(account_data)
+        new_data = {website: {
+            "email": username,
+            "password": password
+        }}
+        print(new_data)
         pyperclip.copy(password)
-
-        # Write information to data file
-        with open("data.txt", "a") as file:
-            file.write(account_data)
+        try:
+            # Write information to data file
+            with open("data.json", "r") as file:
+                # Read/Import json to python dict
+                data = json.load(file)
+        except FileNotFoundError:
+            # Create new data.json file
+            with open("data.json", "w") as file:
+                json.dump(new_data, file, indent=4)
+        else:
+            # If file exists, Update dict to add new data
+            data.update(new_data)
+            with open("data.json", "w") as file:
+                # Add python dict to json
+                json.dump(data, file, indent=4)
+        finally:
             website_input.delete(0, END)
             password_input.delete(0, END)
             display_messagebox("Your info was added successfully!")
@@ -55,6 +71,23 @@ def validate_entries(website, username, password):
         return True
 
 
+def find_password():
+    website = website_input.get()
+    try:
+        with open("data.json", "r") as file:
+            data = json.load(file)
+    except FileNotFoundError:
+        display_messagebox("No data file found.")
+    else:
+        if website in data:
+            email = data[website]['email']
+            password = data[website]['password']
+            display_messagebox(f"Email: {email} \nPassword: {password}")
+            pyperclip.copy(password)
+        else:
+            display_messagebox(f"No details for the {website} exists.")
+
+
 # ---------------------------- UI SETUP ------------------------------- #
 window = Tk()
 window.title("Password Manager")
@@ -69,10 +102,13 @@ def validate_entries(website, username, password):
 website_label = Label(text="Website:")
 website_label.grid(row=1, column=0)
 
-website_input = Entry(width=38)
-website_input.grid(row=1, column=1, columnspan=2)
+website_input = Entry(width=21)
+website_input.grid(row=1, column=1)
 website_input.focus()
 
+website_search = Button(text="Search", width=13, command=find_password)
+website_search.grid(row=1, column=2)
+
 # Username Input
 username_label = Label(text="Email/Username:")
 username_label.grid(row=2, column=0)
@@ -92,7 +128,7 @@ def validate_entries(website, username, password):
 password_btn.grid(row=3, column=2)
 
 # Add to list button
-add_info_btn = Button(text="Add", width=36, command=add_information)
+add_info_btn = Button(text="Add", width=36, command=save)
 add_info_btn.grid(row=4, column=1, columnspan=2)
 
 window.mainloop()