python高级（二）——多任务（二）进程（2）文件夹拷贝器

import os
import multiprocessing


def copy_file(q, file_name, old_folder_name, new_folder_name):
    """完成文件复制"""
    old_f = open(old_folder_name + "/" + file_name)
    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 copy():
    # 1 获取用户要copy的文件夹名字
    old_folder_name = input("要copy的文件夹名：")

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

    # 3 获取文件夹的所有待copy的文件名字
    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=(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()
        # print("完成copy：%s" % file_name)
        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__':
    copy()