多进程copy文件

时间:2022-07-28 15:04:25
from multiprocessing import Pool,Manager 
import os,time
 
def copyFileTask(fileName,oldFolderName,newFolderName,queue):
    fr = open(oldFolderName+"/"+fileName,'r',encoding='UTF-8')
    fw = open(newFolderName+"/"+fileName,'w',encoding='UTF-8')
    #复制
    while True:
        content = fr.read(1024)
        if len(content) == 0:
            break
        fw.write(content)
    #关闭文件
    fr.close()
    fw.close()
 
    queue.put(fileName)
 
 
def main():
    oldFolderName = 'test'#input("请输入要复制的文件夹名字:")
    #创建目录
    newFolderName = oldFolderName+"[复件]"
    os.mkdir(newFolderName)
    #获取文件列表
    fileList = os.listdir(oldFolderName)
    #使用多进程的方式复制
    pool = Pool(5)
    #消息队列
    queue = Manager().Queue()
    for file in fileList:
        #copyFileTask(file,oldFolderName,newFolderName,queue)
        pool.apply_async(copyFileTask,args=(file,oldFolderName,newFolderName,queue))
 
    num = 0
    total = len(fileList)
 
    while num<total:
        queue.get()
        num += 1
        rate = num/total*100
        print('\r复制的进度是:%.2f%%'%rate,end="")
 
    #关闭进程池,不再接受新的进程
    #pool.close()
    #主进程阻塞等待子进程的退出
    #pool.join()
if __name__ == '__main__':
    t_start = time.time()
    main()     
    t_stop = time.time()
    print("\n执行完毕,耗时%0.2f"%(t_stop-t_start))

单进程:

C:\Users\Administrator>python d:\python\copy.py

复制的进度是:100.00%

执行完毕,耗时6.53


多进程:

C:\Users\Administrator>python d:\python\copy.py

复制的进度是:100.00%

执行完毕,耗时6.59