利用多任务完成文件拷贝功能_如何使用多任务方式实现文件的拷贝-优快云博客

本文链接：https://blog.youkuaiyun.com/xiaoshengxihuanfeili/article/details/80113290

本文介绍了一个使用Python多进程进行文件夹内文件复制的例子。通过定义`copy_floder`函数来实现文件从旧文件夹到新文件夹的复制，并通过`generate_pool`函数创建进程池来并行处理文件复制任务。此外，还提供了计算复制进度的功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

python版本：2.7

  #coding=utf-8
  
  from multiprocessing import Pool,Manager      
  import os
  import shutil
  
  def copy_floder(queue,current_road,name,oldFloderName,newFloderName):
      fr = open(current_road+'/'+oldFloderName+'/'+name)
      fw = open(current_road+'/'+newFloderName+'/'+name,'w')
      context = fr.read()
      fw.write(context)
      fr.close()
      fw.close()
  
      queue.put(name)
  
  def generate_pool(queue,current_road,fileNames,oldFloderName,newFloderName):
      pool = Pool(3)
      for name in fileNames:
          pool.apply_async(copy_floder,args=(queue,current_road,name,oldFloderName,newFloderName))
      pool.close()
      pool.join()
  
  def compute_copyRate(fileNames,queue):
      num = 0
      allNum = len(fileNames)
      while num < allNum:
          queue.get()
          num += 1
          copyRate = num / allNum
          print "\rcopy的进度是:%.2f%%"%(copyRate*100),
  
  def main():
      oldFloderName = raw_input("输入文件名:")
      newFloderName = oldFloderName + '副本'
      current_road = os.getcwd()
      all_file = os.listdir(current_road)
  
      if oldFloderName in all_file:
          fileNames = os.listdir(oldFloderName)
          if newFloderName in all_file:
              shutil.rmtree(newFloderName)
              os.mkdir(newFloderName)
          else:
              os.mkdir(newFloderName)
          queue = Manager().Queue()
          generate_pool(queue,current_road,fileNames,oldFloderName,newFloderName)
          compute_copyRate(fileNames,queue)
      else:
          print('该文件不存在，请重试！')
          main()
          
  if __name__ == '__main__':
      main()