拷贝1个文件,我们前面实现过。
过程是:打开文件读出数据data存储,
然后打开另外一个文件写入数据data
就完成了一个文件的拷贝。
首先拷贝一个文件夹,有未知文件数。
所以用多任务拷贝,进程池循环来拷贝
导入模块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 += 1
print ("已经完成了%.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 multiprocessing
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)
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 =0
all_number = len(file_names)
while True:
if count<all_number:
file_name = q.get()
count += 1
print ("已经完成了%.2f %%" % (count*100/all_number))
else:
break
if __name__ == '__main__':
main()
得到的效果: