异常教程

使用 Python 实现一个密码管理器类（手把手讲解）

更新时间:

💡一则或许对你有用的小广告

欢迎加入小哈的星球 ，你将获得：专属的项目实战(已更新的所有项目都能学习) / 1v1 提问 / Java 学习路线 / 学习打卡 / 每月赠书 / 社群讨论

  • 新开坑项目:《Spring AI 项目实战》 正在持续爆肝中，基于 Spring AI + Spring Boot 3.x + JDK 21...， 点击查看 ;
  • 《从零手撸：仿小红书（微服务架构）》 已完结，基于 Spring Cloud Alibaba + Spring Boot 3.x + JDK 17...，点击查看项目介绍 ;演示链接： http://116.62.199.48:7070 ;
  • 《从零手撸：前后端分离博客项目（全栈开发）》 2 期已完结，演示链接： http://116.62.199.48/ ;

截止目前， 星球 内专栏累计输出 90w+ 字，讲解图 3441+ 张，还在持续爆肝中.. 后续还会上新更多项目，目标是将 Java 领域典型的项目都整一波，如秒杀系统, 在线商城, IM 即时通讯，权限管理，Spring Cloud Alibaba 微服务等等，已有 3100+ 小伙伴加入学习 ，欢迎点击围观

前言：为什么需要密码管理器类？

在数字时代，我们每天需要管理的密码数量越来越多——从社交媒体账号到银行应用，从邮箱到云端存储。手动记录或重复使用同一密码不仅效率低下，还可能带来安全风险。密码管理器通过加密技术保护密码安全，同时提供便捷的管理方式，成为现代数字生活的重要工具。

Python 作为一种简洁且功能强大的编程语言，非常适合实现密码管理器的核心功能。通过编写一个密码管理器类，开发者不仅能掌握面向对象编程的精髓，还能学习加密算法、文件操作和异常处理等实用技能。本文将从零开始，逐步讲解如何用 Python 实现一个具备基本功能的密码管理器类，并通过实际案例演示其应用场景。

需求分析与类设计

核心功能梳理

一个基础的密码管理器需要实现以下功能：

  1. 添加密码：为指定网站或服务存储用户名和密码。
  2. 查询密码：根据网站名称获取对应的密码信息。
  3. 删除密码：移除不再需要的密码条目。
  4. 加密与解密：确保密码数据在存储和传输过程中不被窃取。

类结构设计

我们将创建一个名为 PasswordManager 的类，其核心属性和方法如下：

属性/方法功能描述
key用于加密和解密的密钥（Fernet 密钥格式）
encrypted_data存储加密后的密码数据的字典
__init__(self, key)初始化方法，加载密钥并读取加密数据
add_entry(self, site, username, password)添加新的密码条目
get_entry(self, site)根据网站名称查询密码信息
delete_entry(self, site)删除指定网站的密码条目

加密与解密的核心实现

选择加密算法：Fernet 对称加密

密码管理器的核心是数据加密。Python 的 cryptography 库提供了 Fernet 对称加密算法，其特点是使用同一个密钥进行加密和解密，且保证数据的“不可篡改性”。我们可以将其比喻为一把“智能锁”——只有持有正确密钥的人才能打开，且数据在传输过程中不会被篡改。

安装与密钥生成

pip install cryptography

from cryptography.fernet import Fernet

def generate_key():
    return Fernet.generate_key()

key = generate_key()
with open("secret.key", "wb") as key_file:
    key_file.write(key)

加密与解密的代码实现

在 PasswordManager 类中，我们通过以下步骤处理数据：

  1. 使用密钥初始化 Fernet 对象。
  2. 将明文数据转换为字节格式。
  3. 对数据进行加密，生成密文。
  4. 存储密文到文件或内存中。
from cryptography.fernet import Fernet
import json

class PasswordManager:
    def __init__(self, key):
        self.key = key
        self.fernet = Fernet(key)
        self.encrypted_data = {}
        # 从文件加载已加密的数据（后续实现）

数据存储与读取的细节

文件存储方案：JSON 格式

为了持久化存储加密后的数据，我们可以使用 JSON 格式。每个密码条目以字典形式保存，加密后的密码和用户名作为值：

{
    "website1": "gAAAA...",
    "website2": "hBBBBB..."
}

文件读写代码实现

def load_data(self):
    try:
        with open("passwords.json", "r") as f:
            encrypted_data = json.load(f)
        # 将数据转换为字典对象
        self.encrypted_data = encrypted_data
    except FileNotFoundError:
        self.encrypted_data = {}

def save_data(self):
    with open("passwords.json", "w") as f:
        json.dump(self.encrypted_data, f)

加密与存储的完整流程

当用户调用 add_entry 方法时，数据的加密和存储过程如下：

def add_entry(self, site, username, password):
    # 组合明文数据（用户名+密码）
    raw_data = f"{username}:{password}"
    # 加密数据
    encrypted = self.fernet.encrypt(raw_data.encode())
    # 存储到加密数据字典
    self.encrypted_data[site] = encrypted.decode()
    self.save_data()  # 同步到文件

异常处理与用户体验优化

处理常见错误场景

密码管理器需要应对多种异常情况，例如：

  • 密钥文件丢失或损坏
  • 密码条目不存在
  • 用户输入格式错误

示例：密码查询时的异常处理

def get_entry(self, site):
    try:
        encrypted = self.encrypted_data[site]
        decrypted = self.fernet.decrypt(encrypted.encode())
        return decrypted.decode()
    except KeyError:
        return "该网站的密码不存在！"
    except Exception as e:
        return f"发生未知错误：{str(e)}"

用户交互改进

为了提升用户体验，可以添加以下功能：

  • 密钥验证机制（确保密钥正确性）
  • 数据备份与恢复
  • 界面化交互（如命令行提示）

完整代码示例与测试

综合代码实现

from cryptography.fernet import Fernet
import json

class PasswordManager:
    def __init__(self, key):
        self.key = key
        self.fernet = Fernet(key)
        self.encrypted_data = {}
        self.load_data()
    
    def load_data(self):
        try:
            with open("passwords.json", "r") as f:
                self.encrypted_data = json.load(f)
        except FileNotFoundError:
            self.encrypted_data = {}
    
    def save_data(self):
        with open("passwords.json", "w") as f:
            json.dump(self.encrypted_data, f)
    
    def add_entry(self, site, username, password):
        raw_data = f"{username}:{password}"
        encrypted = self.fernet.encrypt(raw_data.encode())
        self.encrypted_data[site] = encrypted.decode()
        self.save_data()
    
    def get_entry(self, site):
        try:
            encrypted = self.encrypted_data[site]
            decrypted = self.fernet.decrypt(encrypted.encode())
            return decrypted.decode()
        except KeyError:
            return "该网站的密码不存在！"
        except Exception as e:
            return f"发生错误：{str(e)}"
    
    def delete_entry(self, site):
        if site in self.encrypted_data:
            del self.encrypted_data[site]
            self.save_data()
            return f"已删除 {site} 的密码"
        else:
            return "该网站的密码不存在！"

测试流程

  1. 生成密钥： 
    key = Fernet.generate_key()
with open("secret.key", "wb") as f:
    f.write(key)
  2. 创建实例并操作： 
    pm = PasswordManager(key)
pm.add_entry("email.com", "user123", "secure_password123")
print(pm.get_entry("email.com"))  # 输出：user123:secure_password123
pm.delete_entry("email.com")

安全性增强建议

密钥的妥善管理

  • 避免硬编码密钥：将密钥保存在安全的文件或环境变量中，而非直接写入代码。
  • 定期更换密钥：通过 generate_key() 生成新密钥，并重新加密所有数据。

数据存储的额外保护

  • 使用更安全的加密算法：如 AES-256 等进阶方案（需调整代码逻辑）。
  • 添加数据完整性校验：通过哈希值验证数据是否被篡改。

结论：从代码到实践

通过本文的讲解，我们实现了一个具备基本功能的密码管理器类。这个项目不仅展示了 Python 在加密和文件操作中的强大能力，还帮助开发者理解了面向对象编程的核心思想。对于初学者而言，这是一个绝佳的实践机会，可以在此基础上进一步扩展功能，例如：

  • 添加图形化界面（如 Tkinter）
  • 支持多用户模式
  • 集成云同步功能

密码管理器的设计体现了“安全与便利”的平衡——通过合理的代码架构和加密技术，我们既能保护敏感信息，又能为用户提供流畅的使用体验。希望本文能激发你对密码管理技术的兴趣，并鼓励你将理论付诸实践。

最新发布

查看更多文章