Python多任务(7.多进程的应用：文件的拷贝器例子)

涉及的知识点

1.    os.mkdir(path)   # os模块的功能，在指定路径上创建文件夹

      os.listdir(path)   #   os模块的功能，打开指定文件夹里面的内容，返回一个列表，该文件夹里面的所有文件都在列表里面

  2.  pool = multiprocessing.Pool(5)  # 创建一个进程池，最大进程数为5

       q = multiprocessing.Manager().Queue()  # 创建一个队列   ，(注意是Manager里面的Queue()方法)

      po.apply_async(要调用的目标,(传递给目标的参数元组,))

      po.close()                     # 关闭进程池，关闭后po不再接受新的请求

 代码如下：

  我要拷贝的文件夹：  D:\shuju\文件Copy器     

'''文件Copy (多进程)'''
import os
import multiprocessing


def copy_file(q, file_name, original_document, new_file):
    '''文件的复制'''
    with open(original_document + "/" + file_name, "rb") as od_f:
        content = od_f.read()

    with open(new_file + "/" + file_name, "wb") as new_f:
        new_f.write(content)

    '''如果拷贝完了文件，那么就向队列中写入一个消息，表示已经完成'''
    q.put(file_name)


def main():
    '''获取用户要copy的文件夹的名字 original_document'''
    original_document = input("请输入原始文件夹的名字：")
    '''2.创建一个新的文件夹 new_file'''
    try:
        new_file = original_document + "(副本)"
        os.mkdir(new_file)
    except Exception:
        pass

    '''获取文件夹的所有待copy的文件名字 (os.listdir())'''
    file_list = os.listdir(original_document)
    '''创建进程池'''
    pool = multiprocessing.Pool(5)

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

    '''复制原文件夹中的文件，到新文件夹中的文件'''
    for file in file_list:
        pool.apply_async(copy_file, args=(q, file, original_document, new_file))

    pool.close()
    all_file_num = len(file_list)  # 所有文件的个数
    copy_complete = 0
    while True:
        q.get()   # 获取文件数据
        copy_complete += 1
        print("\r拷贝的进度：%.f %%" % (copy_complete * 100 / all_file_num),end="")
        if copy_complete >= all_file_num:
            break
    print()  # 换行
    print("拷贝完成")


if __name__ == '__main__':
    main()

 运行结果：

请输入原始文件夹的名字：D:\shuju\文件Copy器
拷贝的进度：100 %
拷贝完成

 打开文件夹就会有：

  

里面的文件一个不少