Python进程实现多任务文件夹的复制

多任务文件夹复制:
1 获取用户要复制的文件夹名字
2 创建一个新的文件夹
3 获取文件夹的所有待拷贝的文件名字
4 创建进程池
5 添加拷贝任务

import multiprocessing
import os


def copy_file(q, file_name, new_folder_name, old_folder_name):
    '''完成文件拷贝'''
    with open(old_folder_name + "/" + file_name, "rb") as f1:
        content = f1.read()
    '''保存到新的文件夹中'''
    with open(new_folder_name + "/" + file_name, "wb") as f2:
        f2.write(content)

    q.put(file_name)

def main():
    '''获取用户要复制的文件夹名字'''
    old_folder_name = input("请输入要复制的文件夹名字:")
    '''创建一个新的文件夹'''
    new_folder_name = old_folder_name + "(.bak文件)"

    if not os.path.exists(new_folder_name):
        os.mkdir(new_folder_name)

    '''获取文件夹的所有待拷贝的文件名字'''
    file_names = os.listdir(old_folder_name)

    '''创建进程池'''
    po = multiprocessing.Pool(5)

    '''创建队列'''
    q = multiprocessing.Manager().Queue()

    '''添加拷贝任务'''
    for file_name in file_names:
        po.apply_async(copy_file, args=(q, file_name, new_folder_name, old_folder_name))

    po.close() 
    po.join()


if __name__ == '__main__':
    main()