一段简单的Python对称加密代码

在看代码之前先把重要的事情说3遍：

代码只有加密部分，请确保你加密的文件有备份，否则后果自负

代码只有加密部分，请确保你加密的文件有备份，否则后果自负

代码只有加密部分，请确保你加密的文件有备份，否则后果自负！！！

脚本效果如下：

加密前

加密后

下面是代码展示：

import os
from cryptography.fernet import Fernet

def encrypt_file(file_path, key):
    # 读取文件内容
    with open(file_path, 'rb') as f:
        data = f.read()

    # 创建 Fernet 对象
    fernet = Fernet(key)

    # 加密数据
    encrypted_data = fernet.encrypt(data)

    # 更新文件名
    file_name, file_ext = os.path.splitext(file_path)
    encrypted_file_path = file_name + file_ext + '.已加密'

    # 写入加密后的数据
    with open(encrypted_file_path, 'wb') as f:
        f.write(encrypted_data)
    #os.remove(file_path)      若想在加密后删除原始文件请把这一行的注释删掉
    # 返回加密后的文件路径
    return encrypted_file_path

# 设置加密密钥
key = Fernet.generate_key()

# 指定文件路径
file_path = 'D:/测试.txt'#这里填上你要加密的文件路径

# 调用函数进行加密
encrypted_file_path = encrypt_file(file_path, key)