多任务案例--文件夹copy.py

import os
import multiprocessing


def copy_file(q,file_name,new_folder_name,old_folder_name):
    with open(old_folder_name + "/" + file_name,"rb") as f1, open(new_folder_name + "/" + file_name,"wb") as f2:
        content = f1.read()
        f2.write(content)
    q.put(file_name)



def main():
    # 1.获取文件夹名字
    old_folder_name = input("请输入想要copy的文件夹的名字：")
    # 2.创建一个新文件夹
    try:
        new_folder_name = old_folder_name + "[复件]"
        os.mkdir(new_folder_name)
    except:
        # 如果创建失败直接pass
        pass
    # 3.获取文件夹里文件的名字
    file_names = os.listdir(old_folder_name)
    # 4.创建进程池
    po = multiprocessing.Pool(5)
    q = multiprocessing.Manager().Queue()
    for file_name in file_names:
        # 5.复制文件到新文件夹里去
        po.apply_async(copy_file,(q,file_name,new_folder_name,old_folder_name))

    po.close()

    # 计算数据
    copy_ok_num = 0
    while True:
        new_file_name = q.get()
        # 记录复制完成的数量
        copy_ok_num+=1
        print("
拷贝的进度为%.2f %%" % (copy_ok_num * 100 / len(file_names)),end="")
        if copy_ok_num >= len(file_names):
            break


if __name__ == '__main__':
    main()

请输入想要copy的文件夹的名字：test
拷贝的进度为100.00 %