Python多进程文件复制

import multiprocessing

import os

import time

import random

def copy_file(queue,file_name,source_folder_name,dest_folder_name):

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()