python多进程拷贝文件

#encoding:utf-8
from multiprocessing import Pool, Manager
import os

def copyFileTask(name, oldFolderName, newFolderName, queue):
    "完成copy一个文件的功能"
    fr = open(oldFolderName + "/" + name)
    fw = open(newFolderName + "/" + name, "w")
    content = fr.read()
    fw.write(content)

    fr.close()
    fw.close()

    queue.put(name)


def main():

    #0:获取要拷贝的文件夹的名字
    oldFolderName = input("请输入文件夹的名字:")

    #1:创建一个文件夹
    newFolderName = oldFolderName + "-复件"
    #print(newFolderName)
    os.mkdir(newFolderName)

    #2:获取old文件夹中的所有的文件名字
    fileNames = os.listdir(oldFolderName)   #返回值为一个列表
    #print(fileNames)

    #3:使用多进程的方式copy原文件夹中的所有文件到新的文件夹中
    pool = Pool(5)  #创建进程池,进程个数为5
    queue = Manager().Queue()

    for name in fileNames:
        pool.apply_async(copyFileTask, args=(name, oldFolderName, newFolderName, queue))

    num = 0
    allNum = len(fileNames)
    while num < allNum:
        queue.get()
        num += 1
        copyRate = num/allNum
        print("\rcopy的进度是:%.2f%%"%(copyRate*100), end = "")

    print("\n已完成copy.....")


if __name__ == "__main__":
    main()

猜你喜欢

转载自blog.csdn.net/duke10/article/details/79869694