python学习-进程池，queue 实例copy文件夹

import os
import multiprocessing

def copy_file(q,file_name,old_folder_name,new_folder_name):
    """完成文件的复制"""
    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")
    #print("new_f"+new_f)
    new_f.write(content)
    new_f.close()
    #print("copy 结束！")
    #copy结束往队列中写一个名称
    q.put(file_name)

def main():
    #1.获取用户要copy的文件夹的名字


    old_folder_name=input("请输入要copy的名字")
    #2.创建一个新的文件夹
    try:

        new_folder_name=old_folder_name+"复件"
        os.rmdir(new_folder_name)
        os.mkdir(new_folder_name)
    except:
        pass
    #3.获取文件夹的所有的待copy的文件名称   listdir()
    file_names=os.listdir(old_folder_name)
    print(file_names)
    #4.多进程，创建进程池
    po = multiprocessing.Pool(5)#括号中的数字为进程池中同时运行的进程数           l
    #5.创建一个队列
    q=multiprocessing.Manager().Queue()
    #6.向进程池添加copy的任务
    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(old_folder_name)
    copy_file_num=0
    #使用队列来显示copy进度
    while True:
        file_name=q.get()
        copy_file_num+=1
        #print("已完成copy："+file_name)
        print("\r拷贝的进度为：%.2f%%" % (copy_file_num*100/all_file_num),end="") #\r为回到打印的行首，end=""不要换行
        if copy_file_num>=all_file_num:
            break



if __name__=="__main__" :
    main()

转载于:https://www.cnblogs.com/quan54/p/10537520.html