基于多进程的文件夹拷贝器

最新推荐文章于 2024-08-12 22:30:43 发布

原创最新推荐文章于 2024-08-12 22:30:43 发布 · 269 阅读

0 ·

CC 4.0 BY-SA版权

Python 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一个使用Python的multiprocessing模块实现的并行文件复制程序。该程序通过创建进程池和队列来加速文件从源文件夹到目标文件夹的复制过程，并实时显示复制进度。

import multiprocessing
import os

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')
    new_f.write(content)
    new_f.close()

    # 如果拷贝完了文件，那么就向队列中写入一个消息，表示已经完成
    q.put(file_name)


def main():
    # 1.获取用户要copy的文件夹的名字
    global new_folder_name
    old_folder_name = input("请输入要copy的文件夹的名字：")
    # 2.创建一个新的文件夹
    try:
        new_folder_name = old_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)
    # 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_name = len(file_names)
    copy_ok_num = 0
    while True:
    	# 如果不将队列中的文件名拿出来，主进程会直接结束
        file_name = q.get()
        # print('已经完成copy： %s' % file_name)
        copy_ok_num += 1
        print("\r拷贝的进度为： %.2f %%" % (copy_ok_num*100/ all_file_name),end='')
        if copy_ok_num >= all_file_name:
            break

if __name__ == '__main__':
    main()