python核心编程:多进程拷贝文件夹下所有文件

'''
多进程拷贝文件夹下所有文件
1.通过输入的路径找到需要被拷贝的文件夹
2.读取文件夹下所有文件名
3.启动进程进行拷贝
4.主进程中显示文件拷贝进度
'''
'''
多进程拷贝文件夹下所有文件
1.通过输入的路径找到需要被拷贝的文件夹
2.读取文件夹下所有文件名
3.启动进程进行拷贝
4.主进程中显示文件拷贝进度
'''
import multiprocessing
import time
import os
import random
from multiprocessing import Pool,Manager

#定义实现拷贝功能
def copy_File(queue,file,oldFolderName,newFolderName):
    fr = open(oldFolderName + '/' + file,'rb')
    fw = open(newFolderName + '/' + file,'wb')
    while True:
        context = fr.read()
        if context:
            fw.write(context)
        else:
            break
    fw.close()
    fr.close()
    time.sleep(random.random())
    queue.put(file)


def main():
    oldFolderName = input('请输入需要复制的文件夹路径:')
    newFolderName = oldFolderName + '-副本'

    #获取文件夹下文件列表
    files = os.listdir(oldFolderName)

    #判断是否存在该文件夹
    if not os.path.exists(newFolderName):
        os.mkdir(newFolderName)
        queue = Manager().Queue()
        pool = Pool(5)

        for file in files:
            pool.apply_async(copy_File, args=(queue, file, oldFolderName, newFolderName))

        pool.close()
        #pool.join()#有了这句话会直接显示完成度100%没有进度过程

        allnum = len(files)
        copynum = 0
        while copynum < allnum:
            file_name = queue.get()
            copynum += 1
            print('\r拷贝进度%.2f%%' % (copynum * 100 / allnum), end='')
        else:
            False

        print('')
    else:
        print('此路径下已存在该文件夹名称,无法完成!')

if __name__ == '__main__':
    main()

在函数里使用了队列的put方法发送了拷贝完的文件名字,之后要用get方法取出才可以进行下一步,否则就相当于上锁后未解锁,是无法成功复制文件的

你可能感兴趣的:(python核心编程)