python中如何實現(xiàn)多任務(wù)copy文件

這篇文章給大家分享的是有關(guān)python中如何實現(xiàn)多任務(wù)copy文件的內(nèi)容。小編覺得挺實用的，因此分享給大家做個參考，一起跟隨小編過來看看吧。

具體如下：

import os
import multiprocessing
def copy_file(q,file_name, old_folder_name, new_folder_name):
  """完成文件的復(fù)制"""
  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()
  # print("從%s--->到%s 復(fù)制文件%s成功！" % (old_folder_name, new_folder_name, file_name))
  q.put(file_name)
def main():
  # 1.獲取用戶要copy的文件夾名稱
  old_folder_name = input("請輸入要copy的文件夾：")
  # 2.創(chuàng)建一個新的文件夾
  try:
    new_folder_name = old_folder_name + "復(fù)件"
    os.mkdir(new_folder_name)
  except:
    pass
  # 3.獲取所有待copy的文件名稱 listdir()
  file_names = os.listdir(old_folder_name)
  print(file_names)
  # 4.創(chuàng)建進(jìn)程池
  po = multiprocessing.Pool(2)
  # 5.創(chuàng)建一個隊列
  q = multiprocessing.Manager().Queue()
  # 6.向進(jìn)程池中添加copy文件的任務(wù)
  for file_name in file_names:
    po.apply_async(copy_file, args=(q,file_name, old_folder_name, new_folder_name))
  po.close()
  # po.join()
  all_file_num = len(file_names) # 統(tǒng)計所有的文件個數(shù)
  copy_ok_num = 0
  while True:
    file_name = q.get()
    # print("已經(jīng)完成copy ：%s" % file_name)
    copy_ok_num += 1
    print("\r復(fù)制的進(jìn)度為：%.2f %%" % (copy_ok_num*100/all_file_num),end='')
    if copy_ok_num >= all_file_num:
      break
if __name__ == '__main__':
  main()

感謝各位的閱讀！關(guān)于“python中如何實現(xiàn)多任務(wù)copy文件”這篇文章就分享到這里了，希望以上內(nèi)容可以對大家有一定的幫助，讓大家可以學(xué)到更多知識，如果覺得文章不錯，可以把它分享出去讓更多的人看到吧！