python怎么批量压缩文件生成密码

Python怎么批量压缩文件生成密码

在日常生活和工作中，我们经常会遇到需要将大量文件进行压缩的情况，如果只是简单地将这些文件压缩成一个压缩包，那么当其他人需要解压这些文件时，就需要知道原始的密码才能进行解压操作，为了保护文件的安全性和隐私性，我们可以为压缩包设置密码，本文将介绍如何使用Python批量压缩文件并生成密码。

准备工作

在使用Python进行文件压缩之前，我们需要先安装两个Python库：zipfile 和 os。zipfile 库用于创建和操作ZIP压缩包，而 os 库则用于处理文件和目录。

打开命令行窗口(Windows)或终端(macOS/Linux),输入以下命令安装这两个库：

pip install zipfile os

编写代码

接下来，我们将编写一个Python脚本来实现批量压缩文件并生成密码的功能，整个过程主要包括以下几个步骤：

1、获取目标文件夹下的所有文件；

2、遍历所有文件，对每个文件进行压缩；

3、为每个压缩包设置密码；

4、将压缩包保存到指定位置。

下面是一个简单的示例代码：

import os
import zipfile
from getpass import getpass
def compress_files(src_dir, dst_dir, password):
    """
    批量压缩文件并生成密码。
    
    参数：
        src_dir (str): 源文件夹路径。
        dst_dir (str): 目标文件夹路径。
        password (str): 压缩包的密码。
    """
     确保目标文件夹存在
    os.makedirs(dst_dir, exist_ok=True)
    
     遍历源文件夹下的所有文件
    for root, dirs, files in os.walk(src_dir):
        for file in files:
             拼接完整的文件路径
            file_path = os.path.join(root, file)
            
             创建ZIP文件名
            zip_name = os.path.splitext(file_path)[0] + '.zip'
            
             将文件添加到ZIP压缩包中
            with zipfile.ZipFile(os.path.join(dst_dir, zip_name), 'w') as zipf:
                zipf.write(file_path)
            
             为ZIP压缩包设置密码
            with zipfile.ZipFile(os.path.join(dst_dir, zip_name), 'a') as zipf:
                zipf.setpassword(bytes(password, 'utf-8'))
                
            print(f'已将{file_path}压缩为{zip_name},密码为{password}')
if __name__ == '__main__':
    src_dir = input('请输入要压缩的文件夹路径：')
    dst_dir = input('请输入压缩包保存的文件夹路径：')
    max_passwd = getpass('请输入最多允许使用的密码个数：') or '5'   如果用户没有输入密码个数，默认为5个密码尝试次数。
    max_tries = max_passwd[0] * max_passwd[1] if len(max_passwd) > 1 else int(max_passwd) * 3 + 10   根据用户输入的密码个数计算最大尝试次数，如果用户只输入了一个数字，那么默认尝试次数为3次；如果用户输入了两个数字，那么默认尝试次数为3*n+10，这是为了避免用户输入过少的密码个数导致暴力破解时间过短的问题。
    max_length = max([len(x) for x in max_passwd]) + 10 if max_length < len(max_passwd) else len(max_passwd) + 10   根据用户输入的最多允许使用的密码个数计算最大密码长度，如果用户输入的密码个数小于等于5,那么默认最大密码长度为5;如果用户输入的密码个数大于5,那么默认最大密码长度为用户输入的最大密码个数加10，这是为了避免用户输入过长的密码导致程序运行时间过长的问题。