使用tqdm跟踪文件写入与处理进度

import os

def iter_files(folder):
    """
    遍历指定文件夹及其子文件夹中的所有文件，
    并生成每个文件的大小和完整路径。
    """
    for root, _, files in os.walk(folder):
        for file in files:
            file_path = os.path.join(root, file)
            try:
                file_size = os.path.getsize(file_path)
                yield file_size, file_path
            except OSError as e:
                print(f"警告: 无法获取文件大小或访问文件 '{file_path}': {e}")
                continue # 跳过无法访问的文件

from tqdm import tqdm

def iter_with_progress(folder):
    """
    为指定文件夹中的文件处理提供tqdm进度条。
    初始化进度条，并为每个文件生成一个更新进度的回调函数。
    """
    # 1. 计算所有文件的总大小作为tqdm的总量
    total_size = sum(s for s, _ in iter_files(folder))

    # 2. 初始化tqdm进度条
    # unit='B' 表示单位是字节
    # total=total_size 设置总进度
    # unit_scale=True 自动缩放单位（B, KB, MB, GB等）
    # unit_divisor=1024 使用1024作为单位换算基数
    progress = tqdm(unit='B',
                    total=total_size,
                    unit_scale=True,
                    unit_divisor=1024,
                    desc="处理文件")

    # 3. 再次遍历文件，并为每个文件提供更新回调
    for size, file_path in iter_files(folder):
        # 定义一个lambda函数作为回调，当调用时更新进度条
        done = lambda current_file_size=size: progress.update(current_file_size)
        yield done, size, file_path

    # 确保循环结束后关闭进度条
    progress.close()

import os
import time
from base64 import b85encode, b85decode # 导入base85编码/解码函数

# 假设的输入目录和输出目录
INPUT_DIR = 'test_input_files'
OUTPUT_DIR = 'test_output_files'

# 创建一些测试文件
def create_dummy_files(directory, num_files=5, avg_size_kb=100):
    if not os.path.exists(directory):
        os.makedirs(directory)
    print(f"创建测试文件于: {directory}")
    for i in range(num_files):
        filename = os.path.join(directory, f"file_{i+1}.txt")
        # 生成随机内容
        content = os.urandom(avg_size_kb * 1024 // 2 + i * 1024) # 稍微变化大小
        with open(filename, 'wb') as f:
            f.write(content)
    print("测试文件创建完成。")

# 确保辅助函数已定义
# (这里省略重复定义 iter_files 和 iter_with_progress，假设它们已在上方定义)
# ... iter_files 和 iter_with_progress 函数定义 ...

def process_files_with_progress(input_folder, output_folder, operation_type='encrypt'):
    """
    带进度条的文件处理函数（加密或解密）。
    """
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)

    print(f"\n开始{operation_type}文件...")
    for done_callback, file_size, input_file_path in iter_with_progress(input_folder):
        relative_path = os.path.relpath(input_file_path, input_folder)
        output_file_path = os.path.join(output_folder, relative_path)

        # 确保输出目录存在
        output_file_dir = os.path.dirname(output_file_path)
        if not os.path.exists(output_file_dir):
            os.makedirs(output_file_dir)

        try:
            print(f"\n正在处理: {relative_path} ({file_size / 1024:.2f} KB)")
            with open(input_file_path, 'rb') as infile:
                original_bytes = infile.read()

            processed_bytes = b''
            if operation_type == 'encrypt':
                processed_bytes = b85encode(original_bytes)
            elif operation_type == 'decrypt':
                processed_bytes = b85decode(original_bytes)
            else:
                raise ValueError("operation_type 必须是 'encrypt' 或 'decrypt'")

            with open(output_file_path, 'wb') as outfile:
                outfile.write(processed_bytes)

            # 模拟处理时间
            # time.sleep(file_size / (1024 * 1024 * 5)) # 模拟每MB处理0.2秒

            done_callback() # 关键：文件处理完成后调用回调函数更新进度条
            print(f"完成处理: {relative_path}")

        except PermissionError:
            print(f"\r跳过 (权限错误): {relative_path}\n")
        except Exception as e:
            print(f"\r处理失败 ({relative_path}): {e}\n")
    print(f"\n所有文件{operation_type}处理完成。")

# --- 运行示例 ---
if __name__ == "__main__":
    # 清理旧目录
    import shutil
    if os.path.exists(INPUT_DIR):
        shutil.rmtree(INPUT_DIR)
    if os.path.exists(OUTPUT_DIR):
        shutil.rmtree(OUTPUT_DIR)

    # 1. 创建测试文件
    create_dummy_files(INPUT_DIR, num_files=10, avg_size_kb=500)

    # 2. 执行加密操作并显示进度
    process_files_with_progress(INPUT_DIR, OUTPUT_DIR + '_encrypted', 'encrypt')

    # 3. 模拟解密操作（可选，需要先有加密文件）
    # 为了演示，我们将加密后的文件作为解密输入
    # process_files_with_progress(OUTPUT_DIR + '_encrypted', OUTPUT_DIR + '_decrypted', 'decrypt')