【Python】多进文件拷贝

用python写的一个多进程之间的文件拷贝程序：

版本1

#__author:bo
#date:2019/3/
import os
import multiprocessing


def copy_file(file_name,old_folder_name,new_folder_name):
    # 拷贝文件
    print("模拟考别文件%s到新文件%s，文件名是%s" % (old_folder_name,new_folder_name,file_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")
    new_f.write(content)
    new_f.close()


def main():
    # 1.获取用户要拷贝的文件夹的名称
    old_folder_name=input("请输入要拷贝的文件夹名称")
    # 2.创建一个新的文件夹
    try:
        new_foler_name=old_folder_name+"[复制文件夹]"
        os.mkdir(new_foler_name)
    except:
        pass
    # 3.获取文件夹中所有的copy的文件名称，lisrdir()
    file_names=os.listdir(old_folder_name)
    print(file_names)
    # 4.创建进程池
    po=multiprocessing.Pool(5)

    # 5.向进程池添加copy文件的任务
    for file_name in file_names:
        po.apply_async(copy_file,args=(file_name,old_folder_name,new_foler_name,))

    # 6.复制原文件夹的到新文件夹中


    po.close()
    po.join()


if __name__ == '__main__':
    main()

版本2

#__author:bo
#date:2019/3/
import os
import multiprocessing


def copy_file(q,file_name,old_folder_name,new_folder_name):
    # 拷贝文件
    # print("模拟考别文件%s到新文件%s，文件名是%s" % (old_folder_name,new_folder_name,file_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")
    new_f.write(content)
    new_f.close()
    # 如果拷贝完，就向队列中放入一个文件名
    q.put(file_name)


def main():
    # 1.获取用户要拷贝的文件夹的名称
    old_folder_name=input("请输入要拷贝的文件夹名称")
    # 2.创建一个新的文件夹
    try:
        new_foler_name=old_folder_name+"[复制文件夹]"
        os.mkdir(new_foler_name)
    except:
        pass
    # 3.获取文件夹中所有的copy的文件名称，lisrdir()
    file_names=os.listdir(old_folder_name)
    print(file_names)
    # 4.创建进程池
    po=multiprocessing.Pool(5)

    # 创建一个队列
    q=multiprocessing.Manager().Queue()



    # 5.向进程池添加copy文件的任务
    for file_name in file_names:
        po.apply_async(copy_file,args=(q,file_name,old_folder_name,new_foler_name,))

    # 6.复制原文件夹的到新文件夹中


    po.close()
    # po.join()

    all_file_nums=len(file_names)  # 拷贝文件的长度
    copy_num=0
    while True:
        file_name=q.get()
        # print("已经完成了copy%s" % file_name)
        copy_num+=1
        print("\r拷贝的进度为%.2f %%" %((copy_num)*100/all_file_nums),end="")
        if copy_num>=all_file_nums:
            break
    print()


if __name__ == '__main__':
    main()