python multiprocessing多进程拷贝文件并且显示进度

import os
import time
import multiprocessing


def copy_file(q, file_name, old_folder_name, new_folder_name):
    """复制文件的任务"""
    # print("模拟拷贝文件:从%s到%s 的文件名字是:%s" % (old_folder_name, new_folder_name, file_name))
    # 拼路径 读取出旧文件
    old_f = open(old_folder_name+"/"+file_name, "rb")
    content = old_f.read()
    old_f.close()

    # 写入到新文件
    new_f = open(new_folder_name+"/"+file_name, "wb")
    new_f.write(content)
    new_f.close()

    # 如果拷贝完成就向队列中写入一个消息表使已经完成
    q.put(file_name)


def main():
    # 1.获取用户要copy的文件夹名字
    old_folder_name = input("请输入需要拷贝的文件夹名字:")

    # 2.创建一个新的文件夹
    try:
        new_folder_name = old_folder_name + "新文件夹"
        os.mkdir(new_folder_name)
    except:
        pass

    # 3.获取文件夹的所有待copy的文件名字 os.listdir(pat)
    file_names = os.listdir(old_folder_name)
    # print(file_names)

    # 4.创建进程池 pool
    po = multiprocessing.Pool(5)

    # 创建一个队列Manger.Queue
    q = multiprocessing.Manager().Queue()

    # 5.向进程池中添加 复制文件的任务
    for file_name in file_names:
        po.apply_async(copy_file, args=(q, file_name, old_folder_name, new_folder_name))

    # 关闭
    po.close()
    # 主进程阻塞,等待子进程的退出, 必须在close或terminate之后使用
    # po.join()

    # 判断拷贝百分比
    all_file_num = len(file_names)  # 计算所有文件个数
    copy_ok = 0
    while True:
        file_name = q.get()
        # print("已经完成copy: %s" % file_name)
        copy_ok += 1
        time.sleep(0.1)
        print("\r拷贝的进度为:%.2f %%" % (copy_ok * 100 / all_file_num), end="")

        if copy_ok >= all_file_num:
            break


if __name__ == '__main__':
    main()

你可能感兴趣的:(python multiprocessing多进程拷贝文件并且显示进度)