用Python创建一个简单的密码管理器-CSDN博客

from cryptography.fernet import Fernet

def generate_key():
    """生成一个新的加密密钥"""
    key = Fernet.generate_key()
    with open("secret.key", "wb") as key_file:
        key_file.write(key)

def load_key():
    """加载之前生成的加密密钥"""
    return open("secret.key", "rb").read()

def encrypt_message(message, key):
    """加密消息"""
    f = Fernet(key)
    encrypted_message = f.encrypt(message.encode())
    return encrypted_message

def decrypt_message(encrypted_message, key):
    """解密消息"""
    f = Fernet(key)
    decrypted_message = f.decrypt(encrypted_message).decode()
    return decrypted_message

2. 数据库操作

接下来，我们需要编写一些函数来操作数据库。我们将使用sqlite3库。

import sqlite3

def create_database():
    """创建用于存储密码的数据库"""
    conn = sqlite3.connect("database.db")
    cursor = conn.cursor()
    cursor.execute("""
    CREATE TABLE IF NOT EXISTS passwords (
        id INTEGER PRIMARY KEY,
        service TEXT NOT NULL,
        username TEXT NOT NULL,
        password TEXT NOT NULL
    )
    """)
    conn.commit()
    conn.close()

def add_password(service, username, password):
    """向数据库添加新的密码"""
    conn = sqlite3.connect("database.db")
    cursor = conn.cursor()
    cursor.execute("INSERT INTO passwords (service, username, password) VALUES (?, ?, ?)", (service, username, password))
    conn.commit()
    conn.close()

def get_password(service):
    """从数据库获取密码"""
    conn = sqlite3.connect("database.db")
    cursor = conn.cursor()
    cursor.execute("SELECT username, password FROM passwords WHERE service=?", (service,))
    result = cursor.fetchone()
    conn.close()
    return result

3. 主程序

最后，我们将编写主程序来整合上述功能。用户可以选择添加新的密码或获取已存储的密码。

import os

def main():
    if not os.path.exists("secret.key"):
        generate_key()
    key = load_key()

    create_database()

    while True:
        print("1. 添加新密码")
        print("2. 获取已存储的密码")
        print("3. 退出")
        choice = input("请选择一个操作: ")

        if choice == "1":
            service = input("请输入服务名称: ")
            username = input("请输入用户名: ")
            password = input("请输入密码: ")
            encrypted_password = encrypt_message(password, key)
            add_password(service, username, encrypted_password)
            print("密码已添加!")

        elif choice == "2":
            service = input("请输入服务名称: ")
            result = get_password(service)
            if result:
                username, encrypted_password = result
                password = decrypt_message(encrypted_password, key)
                print(f"用户名: {username}")
                print(f"密码: {password}")
            else:
                print("未找到该服务的密码")

        elif choice == "3":
            break

        else:
            print("无效的选择，请重新选择")

if __name__ == "__main__":
    main()