多进程文件夹copy

# coding=uutf-8
import os
import time
import random
import multiprocessing


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():
    # 1. 获取用户要copy的文件夹名称
    old_folder_name = input("请输入要copy的文件夹名称：")
    # 2. 创建一个新的文件夹
    try:
        new_folder_name = old_folder_name + “[复件]”
        os.mkdir(new_folder_name)
    except:
        pass
    # 3. 获取文件夹的所有待copy的文件名称 listdir()
    file_names = os.listdir(old_folder_name)
    print(file_names)
    # 4. 创建进程池
    po = multiprocessing.Pool(5)
    # 5. 创建一个队列
    q = multiprocessing.Manager().Queue()
    # 6. 向进程池中添加 copy 文件的任务
    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_num = len(file_names) # 获取文件夹内所有文件数量
    copy_ok_num = 0
    while True:
        file_name = q.get()
        copy_ok_num += 1
        print("\r拷贝进度为：%.2f %%"%(copy_ok_num*100/all_file_num,), end = "")
        if copy_ok_num >= all_file_num:
            break
        print()


if __name__ == "__main__":
    main()