python进程池实现的多进程文件夹copy器完整示例

本文实例讲述了python进程池实现的多进程文件夹copy器。分享给大家供大家参考，具体如下：

应用：文件夹copy器（多进程版）

import multiprocessing
import os
import time
import random
def copy_file(queue, file_name,source_folder_name, dest_folder_name):
 """copy文件到指定的路径"""
 f_read = open(source_folder_name + "/" + file_name, "rb")
 f_write = open(dest_folder_name + "/" + file_name, "wb")
 while True:
   time.sleep(random.random())
   content = f_read.read(1024)
   if content:
     f_write.write(content)
   else:
     break
 f_read.close()
 f_write.close()
 # 发送已经拷贝完毕的文件名字
 queue.put(file_name)
def main():
 # 获取要复制的文件夹
 source_folder_name = input("请输入要复制文件夹名字:")
 # 整理目标文件夹
 dest_folder_name = source_folder_name + "[副本]"
 # 创建目标文件夹
 try:
   os.mkdir(dest_folder_name)
 except:
   pass # 如果文件夹已经存在，那么创建会失败
 # 获取这个文件夹中所有的普通文件名
 file_names = os.listdir(source_folder_name)
 # 创建Queue
 queue = multiprocessing.Manager().Queue()
 # 创建进程池
 pool = multiprocessing.Pool(3)
 for file_name in file_names:
   # 向进程池中添加任务
   pool.apply_async(copy_file, args=(queue, file_name, source_folder_name, dest_folder_name))
 # 主进程显示进度
 pool.close()
 all_file_num = len(file_names)
 while True:
   file_name = queue.get()
   if file_name in file_names:
     file_names.remove(file_name)
   copy_rate = (all_file_num-len(file_names))*100/all_file_num
   print("\r％.2f...(％s)" ％ (copy_rate, file_name) + " "*50, end="")
   if copy_rate >= 100:
     break
 print()
if __name__ == "__main__":
 main()

希望本文所述对大家Python程序设计有所帮助。

来源：https://blog.csdn.net/lislislislislis/article/details/89318228