简单的多任务copy_folder

import os
import multiprocessing


import time




def copy_file(q, file_name, old_folder_name, new_folder_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():
    # 获取要复制的文件夹的名字
    old_folder_name = input("输入要复制的文件夹的名字：")


    # 新建一个新的文件夹
    try:
        new_folder_name = old_folder_name + "[副本]"
        os.mkdir(new_folder_name)
    except:
        pass
    # 获取要下载的文件夹中文件的名字  listdir()
    file_names = os.listdir(old_folder_name)
    # print(file_names)


    # 创建进程池,队列
    po = multiprocessing.Pool(5)
    q = multiprocessing.Manager().Queue()


    # 把下载的任务添加到进程池中
    for file_name in file_names:
        po.apply_async(copy_file, args=(q, file_name, old_folder_name, new_folder_name))


    po.close()
    # po.join()
    all_file = len(file_names)
    copy_finish = 0
    while True:
        file_name = q.get()
        copy_finish += 1
        print("\r进度为%.2f %%" % (copy_finish * 100 /all_file), end="")
        time.sleep(0.2)
        # print("")
        if copy_finish >= all_file:
            break


if __name__ == '__main__':
    main()