'''
多进程拷贝文件夹下所有文件
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方法取出才可以进行下一步,否则就相当于上锁后未解锁,是无法成功复制文件的