使用多进程（multiprocess）进行文件夹下文件的拷贝

使用多进程（multiprocess）进行文件夹下文件的拷贝

import os 
import multiprocessing # 多多进程的python 包


def copy_file(q, source_file, old_fold_name, des_folder):
    # 完成文件复制
    # print('---------' + source_file)
    old_file = open(old_fold_name + '/' + source_file, "rb")
    content = old_file.read()
    old_file.close()

    new_file = open(des_folder + '/' + source_file, 'wb')
    new_file.write(content)
    new_file.close()
    q.put(source_file)

    pass


def main():
    # 1.获取用户想要复制的文件夹的名字
    old_folder_name = input("请输入要复制的文件夹的名字:")
    # 2. 创建一个新的文件夹
    try:
        os.mkdir(old_folder_name + "[copy]")
    except:
        pass
    # 3. 获取文件夹的所有的代copy的文件夹的名字 listdir()
    file_names = os.listdir(old_folder_name)
    # print(file_names)
    # 4. 创建进程池，最多同时执行5个进程
    po = multiprocessing.Pool(5)

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

    # 5. 向进程池中添加copy文件的任务
    for file_name in file_names:
        po.apply_async(copy_file, args=(q, file_name, old_folder_name, old_folder_name + "[copy]"))

    po.close()
    # po.join() # 让主线程等待，防止主进程结束导致子进程结束
    file_num = len(file_names)
    copy_ok_num = 0
    # 下面的while用来显示进度，属于附加功能
    while True:
        file_name = q.get()
        copy_ok_num += 1
        print("\r完成进度:%.2f %%" % (copy_ok_num * 100 / file_num),end='')
        if copy_ok_num >= file_num:
            break


if __name__ == "__main__":
    main()