Add warning delete

main.py

@@ -17,6 +17,7 @@ def main():
           4. Load an existing password file
           5. Add a password
           6. Get a password
+          10. Delete a password
           q. Quit
           """)
 
@@ -42,6 +43,13 @@ def main():
         elif choice == '6':
             site = input("Enter site: ").strip()
             print(f"Password for {site}: {pm.get_password(site)}")
+        elif choice == '10':
+            site = input("Enter site to delete password: ").strip()
+            prompt = input(f"Confirm deleting password for {site} ? (y/n)").strip().lower()
+            if(prompt=="y"):
+                pm.delete_password(site)
+            else:
+                print("No changes made!")
         elif choice == 'q':
             done = True
             print("Goodbye!")

manager.py

@@ -39,3 +39,17 @@ def add_password(self, site, password):
 
     def get_password(self, site):
         return self.password_dict.get(site, "Password not found.")
+    def delete_password(self, site):
+        if site in self.password_dict:
+            # Remove from the dictionary
+            del self.password_dict[site]
+            # Update the password file
+            if self.password_file is not None:
+                with open(self.password_file, 'r') as f:
+                    lines = f.readlines()
+                with open(self.password_file, 'w') as f:
+                    for line in lines:
+                        if not line.startswith(site + ":"):  # Skip the line with the site to delete
+                            f.write(line)
+        else:
+            print("Site not found.")