进程5 案例：文件夹copy器（多进程版）

拷贝1个文件，我们前面实现过。

过程是：打开文件读出数据data存储，
然后打开另外一个文件写入数据data
就完成了一个文件的拷贝。

• 试想一个文件夹的拷贝又是怎么样的

首先拷贝一个文件夹，有未知文件数。
所以用多任务拷贝，进程池循环来拷贝

导入模块os,multiprocessing
第一步：获取要拷贝的文件夹名字
第二步：获取文件夹中所有要复制的文件名字
第三步：创建一个新的文件夹
第四步：创建进程池
第五步：创建队列
第六步：向进程池中添加复制文件的任务
第七步：复制完从队列中依次输出

• 导入模块os,multiprocessing

os用来提取文件夹中所有文件名，mutiprocessing用来创建进程池，队列

import os
import multiprocessing

• 第一步：获取要拷贝的文件夹名字

old_folder_name = input("请输入要copy的文件夹的名字：")

• 第二步：获取文件夹中所有要复制的文件名字

 file_names = os.listdir(old_folder_name)

• 第三步：创建一个新的文件夹

创建一个新的文件夹，如果有就不创建

try:new_folder_name = old_folder_name + "[附件]"os.mkdir(new_folder_name)
except:pass

• 第四步：创建进程池

其中的进程完成复制的任务

 po = multiprocessing.Pool(5)

• 第五步：创建队列

队列通信完成主进程显示进度

q = multiprocessing.Manager().Queue()

• 第六步：向进程池中添加复制文件的任务

for file_name in file_names:po.apply_async(copy_file,args=(q,file_name,old_folder_name,new_folder_name))

• 第七步：复制完从队列中依次输出

count =0
all_number = len(file_names)
while True:if count<all_number:file_name = q.get()count += 1print ("已经完成了%.2f %%" % (count*100/all_number))else:break

• 复制文件的方法

把此方法调入到进程池中进行

def copy_file(q,file_name,old_folder_name,new_folder_name):"""完成文件的复制"""old_f = open(old_folder_name + "/" + file_name, "rb")content = old_f.read()old_f.close()new_f = open(new_folder_name + "/" + file_name, "wb")new_f.write(content)new_f.close()#如果拷贝完了文件，向队列中写入一个消息，表示已经完成q.put(file_name)

总代码:

import os
import multiprocessingdef copy_file(q,file_name,old_folder_name,new_folder_name):"""完成文件的复制"""old_f = open(old_folder_name + "/" + file_name, "rb")content = old_f.read()old_f.close()new_f = open(new_folder_name + "/" + file_name, "wb")new_f.write(content)new_f.close()#如果拷贝完了文件，向队列中写入一个消息，表示已经完成q.put(file_name)def main():# 1.获取用户要copy的文件夹的名字old_folder_name = input("请输入要copy的文件夹的名字：")# 2.创建一个新的文件夹try:new_folder_name = old_folder_name + "[附件]"os.mkdir(new_folder_name)except:pass# 3.获取文件夹中所有的待copy的文件名字  listdir()file_names = os.listdir(old_folder_name)# 4.创建进程池 主要负责复制文件po = multiprocessing.Pool(5)# 5.创建一个队列q = multiprocessing.Manager().Queue()# 6.向进程池中添加copy文件的任务for file_name in file_names:po.apply_async(copy_file,args=(q,file_name,old_folder_name,new_folder_name))po.close()count =0all_number = len(file_names)while True:if count<all_number:file_name = q.get()count += 1print ("已经完成了%.2f %%" % (count*100/all_number))else:breakif __name__ == '__main__':main()

得到的效果：

本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！