python進(jìn)程池如何實(shí)現(xiàn)多進(jìn)程文件夾copy器

這篇文章主要介紹了python進(jìn)程池如何實(shí)現(xiàn)多進(jìn)程文件夾copy器，具有一定借鑒價(jià)值，感興趣的朋友可以參考下，希望大家閱讀完這篇文章之后大有收獲，下面讓小編帶著大家一起了解一下。

應(yīng)用：文件夾copy器（多進(jìn)程版）

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()
  # 發(fā)送已經(jīng)拷貝完畢的文件名字
  queue.put(file_name)
def main():
  # 獲取要復(fù)制的文件夾
  source_folder_name = input("請(qǐng)輸入要復(fù)制文件夾名字:")
  # 整理目標(biāo)文件夾
  dest_folder_name = source_folder_name + "[副本]"
  # 創(chuàng)建目標(biāo)文件夾
  try:
    os.mkdir(dest_folder_name)
  except:
    pass # 如果文件夾已經(jīng)存在，那么創(chuàng)建會(huì)失敗
  # 獲取這個(gè)文件夾中所有的普通文件名
  file_names = os.listdir(source_folder_name)
  # 創(chuàng)建Queue
  queue = multiprocessing.Manager().Queue()
  # 創(chuàng)建進(jìn)程池
  pool = multiprocessing.Pool(3)
  for file_name in file_names:
    # 向進(jìn)程池中添加任務(wù)
    pool.apply_async(copy_file, args=(queue, file_name, source_folder_name, dest_folder_name))
  # 主進(jìn)程顯示進(jìn)度
  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()

感謝你能夠認(rèn)真閱讀完這篇文章，希望小編分享的“python進(jìn)程池如何實(shí)現(xiàn)多進(jìn)程文件夾copy器”這篇文章對(duì)大家有幫助，同時(shí)也希望大家多多支持億速云，關(guān)注億速云行業(yè)資訊頻道，更多相關(guān)知識(shí)等著你來學(xué)習(xí)!