Password Manager

# Code for Password Generator

import os
import json
from cryptography.fernet import Fernet, InvalidToken

# Functions for password manager
def generate_key():
    return Fernet.generate_key()

def create_fernet(key):
    return Fernet(key)

def encrypt_password(fernet, password):
    return fernet.encrypt(password.encode())

def decrypt_password(fernet, encrypted_password):
    return fernet.decrypt(encrypted_password).decode()

def load_passwords(fernet, filename):
    try:
        with open(filename, 'r') as file:
            data = json.load(file)
        return {key: decrypt_password(fernet, value) for key, value in data.items()}
    except (FileNotFoundError, InvalidToken):
        return {}

def save_passwords(fernet, passwords, filename):
    data = {key: encrypt_password(fernet, value) for key, value in passwords.items()}
    with open(filename, 'w') as file:
        json.dump(data, file)

def add_password(fernet, passwords, account, password, filename):
    passwords[account] = password
    save_passwords(fernet, passwords, filename)

def update_password(fernet, passwords, account, password, filename):
    if account in passwords:
        passwords[account] = password
        save_passwords(fernet, passwords, filename)
    else:
        print(f"Account '{account}' not found.")

def delete_password(fernet, passwords, account, filename):
    if account in passwords:
        del passwords[account]
        save_passwords(fernet, passwords, filename)
    else:
        print(f"Account '{account}' not found.")

# Main function
def main():
    key = input("Enter the master password (keep this safe!): ").encode()
    fernet = create_fernet(key)
    filename = "passwords.json"

    passwords = load_passwords(fernet, filename)

    while True:
        print("\nOptions:")
        print("1. Show all accounts")
        print("2. Retrieve password")
        print("3. Add account")
        print("4. Update account password")
        print("5. Delete account")
        print("6. Quit")

        choice = input("Enter your choice: ")

        if choice == "1":
            print("Accounts:")
            for account in passwords.keys():
                print(account)
        elif choice == "2":
            account = input("Enter the account name: ")
            if account in passwords:
                print(f"Password for {account}: {passwords[account]}")
            else:
                print(f"Account '{account}' not found.")
        elif choice == "3":
            account = input("Enter the account name: ")
            password = input("Enter the password: ")
            add_password(fernet, passwords, account, password, filename)
            print(f"Account '{account}' added.")
        elif choice == "4":
            account = input("Enter the account name: ")
            password = input("Enter the new password: ")
            update_password(fernet, passwords, account, password, filename)
        elif choice == "5":
            account = input("Enter the account name: ")
            delete_password(fernet, passwords, account, filename)
        elif choice == "6":
            break
        else:
            print("Invalid choice. Please try again.")

if __name__ == "__main__":
    main()